faiss 0.3.0 → 0.3.2

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: faiss
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.2
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-12 00:00:00.000000000 Z
+date: 2024-10-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -107,6 +107,8 @@ files:
 - vendor/faiss/faiss/IndexIVFFastScan.h
 - vendor/faiss/faiss/IndexIVFFlat.cpp
 - vendor/faiss/faiss/IndexIVFFlat.h
+- vendor/faiss/faiss/IndexIVFIndependentQuantizer.cpp
+- vendor/faiss/faiss/IndexIVFIndependentQuantizer.h
 - vendor/faiss/faiss/IndexIVFPQ.cpp
 - vendor/faiss/faiss/IndexIVFPQ.h
 - vendor/faiss/faiss/IndexIVFPQFastScan.cpp
@@ -123,6 +125,8 @@ files:
 - vendor/faiss/faiss/IndexNNDescent.h
 - vendor/faiss/faiss/IndexNSG.cpp
 - vendor/faiss/faiss/IndexNSG.h
+- vendor/faiss/faiss/IndexNeuralNetCodec.cpp
+- vendor/faiss/faiss/IndexNeuralNetCodec.h
 - vendor/faiss/faiss/IndexPQ.cpp
 - vendor/faiss/faiss/IndexPQ.h
 - vendor/faiss/faiss/IndexPQFastScan.cpp
@@ -153,6 +157,8 @@ files:
 - vendor/faiss/faiss/cppcontrib/SaDecodeKernels.h
 - vendor/faiss/faiss/cppcontrib/detail/CoarseBitType.h
 - vendor/faiss/faiss/cppcontrib/detail/UintReader.h
+- vendor/faiss/faiss/cppcontrib/factory_tools.cpp
+- vendor/faiss/faiss/cppcontrib/factory_tools.h
 - vendor/faiss/faiss/cppcontrib/sa_decode/Level2-avx2-inl.h
 - vendor/faiss/faiss/cppcontrib/sa_decode/Level2-inl.h
 - vendor/faiss/faiss/cppcontrib/sa_decode/Level2-neon-inl.h
@@ -171,6 +177,7 @@ files:
 - vendor/faiss/faiss/gpu/GpuIcmEncoder.h
 - vendor/faiss/faiss/gpu/GpuIndex.h
 - vendor/faiss/faiss/gpu/GpuIndexBinaryFlat.h
+- vendor/faiss/faiss/gpu/GpuIndexCagra.h
 - vendor/faiss/faiss/gpu/GpuIndexFlat.h
 - vendor/faiss/faiss/gpu/GpuIndexIVF.h
 - vendor/faiss/faiss/gpu/GpuIndexIVFFlat.h
@@ -202,6 +209,7 @@ files:
 - vendor/faiss/faiss/gpu/test/TestUtils.h
 - vendor/faiss/faiss/gpu/test/demo_ivfpq_indexing_gpu.cpp
 - vendor/faiss/faiss/gpu/utils/DeviceUtils.h
+- vendor/faiss/faiss/gpu/utils/RaftUtils.h
 - vendor/faiss/faiss/gpu/utils/StackDeviceMemory.cpp
 - vendor/faiss/faiss/gpu/utils/StackDeviceMemory.h
 - vendor/faiss/faiss/gpu/utils/StaticUtils.h
@@ -244,10 +252,11 @@ files:
 - vendor/faiss/faiss/impl/ThreadedIndex-inl.h
 - vendor/faiss/faiss/impl/ThreadedIndex.h
 - vendor/faiss/faiss/impl/code_distance/code_distance-avx2.h
+- vendor/faiss/faiss/impl/code_distance/code_distance-avx512.h
 - vendor/faiss/faiss/impl/code_distance/code_distance-generic.h
 - vendor/faiss/faiss/impl/code_distance/code_distance.h
-- vendor/faiss/faiss/impl/code_distance/code_distance_avx512.h
 - vendor/faiss/faiss/impl/index_read.cpp
+- vendor/faiss/faiss/impl/index_read_utils.h
 - vendor/faiss/faiss/impl/index_write.cpp
 - vendor/faiss/faiss/impl/io.cpp
 - vendor/faiss/faiss/impl/io.h
@@ -261,6 +270,8 @@ files:
 - vendor/faiss/faiss/impl/pq4_fast_scan.h
 - vendor/faiss/faiss/impl/pq4_fast_scan_search_1.cpp
 - vendor/faiss/faiss/impl/pq4_fast_scan_search_qbs.cpp
+- vendor/faiss/faiss/impl/residual_quantizer_encode_steps.cpp
+- vendor/faiss/faiss/impl/residual_quantizer_encode_steps.h
 - vendor/faiss/faiss/impl/simd_result_handlers.h
 - vendor/faiss/faiss/index_factory.cpp
 - vendor/faiss/faiss/index_factory.h
@@ -280,6 +291,8 @@ files:
 - vendor/faiss/faiss/utils/AlignedTable.h
 - vendor/faiss/faiss/utils/Heap.cpp
 - vendor/faiss/faiss/utils/Heap.h
+- vendor/faiss/faiss/utils/NeuralNet.cpp
+- vendor/faiss/faiss/utils/NeuralNet.h
 - vendor/faiss/faiss/utils/WorkerThread.cpp
 - vendor/faiss/faiss/utils/WorkerThread.h
 - vendor/faiss/faiss/utils/approx_topk/approx_topk.h
@@ -287,6 +300,7 @@ files:
 - vendor/faiss/faiss/utils/approx_topk/generic.h
 - vendor/faiss/faiss/utils/approx_topk/mode.h
 - vendor/faiss/faiss/utils/approx_topk_hamming/approx_topk_hamming.h
+- vendor/faiss/faiss/utils/bf16.h
 - vendor/faiss/faiss/utils/distances.cpp
 - vendor/faiss/faiss/utils/distances.h
 - vendor/faiss/faiss/utils/distances_fused/avx512.cpp
@@ -299,6 +313,7 @@ files:
 - vendor/faiss/faiss/utils/extra_distances-inl.h
 - vendor/faiss/faiss/utils/extra_distances.cpp
 - vendor/faiss/faiss/utils/extra_distances.h
+- vendor/faiss/faiss/utils/fp16-arm.h
 - vendor/faiss/faiss/utils/fp16-fp16c.h
 - vendor/faiss/faiss/utils/fp16-inl.h
 - vendor/faiss/faiss/utils/fp16.h
@@ -313,17 +328,21 @@ files:
 - vendor/faiss/faiss/utils/ordered_key_value.h
 - vendor/faiss/faiss/utils/partitioning.cpp
 - vendor/faiss/faiss/utils/partitioning.h
+- vendor/faiss/faiss/utils/prefetch.h
 - vendor/faiss/faiss/utils/quantize_lut.cpp
 - vendor/faiss/faiss/utils/quantize_lut.h
 - vendor/faiss/faiss/utils/random.cpp
 - vendor/faiss/faiss/utils/random.h
 - vendor/faiss/faiss/utils/simdlib.h
 - vendor/faiss/faiss/utils/simdlib_avx2.h
+- vendor/faiss/faiss/utils/simdlib_avx512.h
 - vendor/faiss/faiss/utils/simdlib_emulated.h
 - vendor/faiss/faiss/utils/simdlib_neon.h
+- vendor/faiss/faiss/utils/simdlib_ppc64.h
 - vendor/faiss/faiss/utils/sorting.cpp
 - vendor/faiss/faiss/utils/sorting.h
 - vendor/faiss/faiss/utils/transpose/transpose-avx2-inl.h
+- vendor/faiss/faiss/utils/transpose/transpose-avx512-inl.h
 - vendor/faiss/faiss/utils/utils.cpp
 - vendor/faiss/faiss/utils/utils.h
 homepage: https://github.com/ankane/faiss-ruby
@@ -345,7 +364,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.5.16
 signing_key:
 specification_version: 4
 summary: Efficient similarity search and clustering for Ruby

data/vendor/faiss/faiss/impl/code_distance/code_distance_avx512.h

@@ -1,102 +0,0 @@
-/**
- * Copyright (c) Facebook, Inc. and its affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
-
-// // // AVX-512 version. It is not used, but let it be for the future
-// // // needs.
-// // template <class SearchResultType, typename T = PQDecoder>
-// // typename std::enable_if<(std::is_same<T, PQDecoder8>::value), void>::
-// //         type distance_four_codes(
-// //     const uint8_t* __restrict code0,
-// //     const uint8_t* __restrict code1,
-// //     const uint8_t* __restrict code2,
-// //     const uint8_t* __restrict code3,
-// //     float& result0,
-// //     float& result1,
-// //     float& result2,
-// //     float& result3
-// // ) const {
-// //     result0 = 0;
-// //     result1 = 0;
-// //     result2 = 0;
-// //     result3 = 0;
-
-// //     size_t m = 0;
-// //     const size_t pqM16 = pq.M / 16;
-
-// //     constexpr intptr_t N = 4;
-
-// //     const float* tab = sim_table;
-
-// //     if (pqM16 > 0) {
-// //         // process 16 values per loop
-// //         const __m512i ksub = _mm512_set1_epi32(pq.ksub);
-// //         __m512i offsets_0 = _mm512_setr_epi32(0, 1, 2, 3, 4, 5, 6, 7,
-// //              8, 9, 10, 11, 12, 13, 14, 15);
-// //         offsets_0 = _mm512_mullo_epi32(offsets_0, ksub);
-
-// //         // accumulators of partial sums
-// //         __m512 partialSums[N];
-// //         for (intptr_t j = 0; j < N; j++) {
-// //             partialSums[j] = _mm512_setzero_ps();
-// //         }
-
-// //         // loop
-// //         for (m = 0; m < pqM16 * 16; m += 16) {
-// //             // load 16 uint8 values
-// //             __m128i mm1[N];
-// //             mm1[0] = _mm_loadu_si128((const __m128i_u*)(code0 + m));
-// //             mm1[1] = _mm_loadu_si128((const __m128i_u*)(code1 + m));
-// //             mm1[2] = _mm_loadu_si128((const __m128i_u*)(code2 + m));
-// //             mm1[3] = _mm_loadu_si128((const __m128i_u*)(code3 + m));
-
-// //             // process first 8 codes
-// //             for (intptr_t j = 0; j < N; j++) {
-// //                 // convert uint8 values (low part of __m128i) to int32
-// //                 // values
-// //                 const __m512i idx1 = _mm512_cvtepu8_epi32(mm1[j]);
-
-// //                 // add offsets
-// //                 const __m512i indices_to_read_from =
-// //                     _mm512_add_epi32(idx1, offsets_0);
-
-// //                 // gather 8 values, similar to 8 operations of
-// // //                    tab[idx]
-// //                 __m512 collected =
-// //                        _mm512_i32gather_ps(
-// //                             indices_to_read_from, tab, sizeof(float));
-
-// //                 // collect partial sums
-// //                 partialSums[j] = _mm512_add_ps(partialSums[j],
-// //                    collected);
-// //             }
-// //             tab += pq.ksub * 16;
-
-// //         }
-
-// //         // horizontal sum for partialSum
-// //         result0 += _mm512_reduce_add_ps(partialSums[0]);
-// //         result1 += _mm512_reduce_add_ps(partialSums[1]);
-// //         result2 += _mm512_reduce_add_ps(partialSums[2]);
-// //         result3 += _mm512_reduce_add_ps(partialSums[3]);
-// //     }
-
-// //     //
-// //     if (m < pq.M) {
-// //         // process leftovers
-// //         PQDecoder decoder0(code0 + m, pq.nbits);
-// //         PQDecoder decoder1(code1 + m, pq.nbits);
-// //         PQDecoder decoder2(code2 + m, pq.nbits);
-// //         PQDecoder decoder3(code3 + m, pq.nbits);
-// //         for (; m < pq.M; m++) {
-// //             result0 += tab[decoder0.decode()];
-// //             result1 += tab[decoder1.decode()];
-// //             result2 += tab[decoder2.decode()];
-// //             result3 += tab[decoder3.decode()];
-// //             tab += pq.ksub;
-// //         }
-// //     }
-// // }