faiss 0.5.3 → 0.6.1

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: faiss
 version: !ruby/object:Gem::Version
-  version: 0.5.3
+  version: 0.6.1
 platform: ruby
 authors:
 - Andrew Kane
@@ -15,28 +15,28 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '4.8'
+        version: 4.11.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '4.8'
+        version: 4.11.5
 - !ruby/object:Gem::Dependency
-  name: numo-narray
+  name: numo-narray-alt
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.10'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.10'
 email: andrew@ankane.org
 executables: []
 extensions:
@@ -158,15 +158,17 @@ files:
 - vendor/faiss/faiss/MetaIndexes.cpp
 - vendor/faiss/faiss/MetaIndexes.h
 - vendor/faiss/faiss/MetricType.h
+- vendor/faiss/faiss/SuperKMeans.cpp
+- vendor/faiss/faiss/SuperKMeans.h
 - vendor/faiss/faiss/VectorTransform.cpp
 - vendor/faiss/faiss/VectorTransform.h
+- vendor/faiss/faiss/build.cpp
+- vendor/faiss/faiss/build.h
 - vendor/faiss/faiss/clone_index.cpp
 - vendor/faiss/faiss/clone_index.h
 - 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
@@ -175,6 +177,8 @@ files:
 - vendor/faiss/faiss/cppcontrib/sa_decode/PQ-avx2-inl.h
 - vendor/faiss/faiss/cppcontrib/sa_decode/PQ-inl.h
 - vendor/faiss/faiss/cppcontrib/sa_decode/PQ-neon-inl.h
+- vendor/faiss/faiss/factory_tools.cpp
+- vendor/faiss/faiss/factory_tools.h
 - vendor/faiss/faiss/gpu/GpuAutoTune.cpp
 - vendor/faiss/faiss/gpu/GpuAutoTune.h
 - vendor/faiss/faiss/gpu/GpuCloner.cpp
@@ -227,24 +231,39 @@ files:
 - vendor/faiss/faiss/gpu/utils/StaticUtils.h
 - vendor/faiss/faiss/gpu/utils/Timer.cpp
 - vendor/faiss/faiss/gpu/utils/Timer.h
+- vendor/faiss/faiss/gpu_metal/GpuIndexFlat.h
+- vendor/faiss/faiss/gpu_metal/MetalCloner.h
+- vendor/faiss/faiss/gpu_metal/MetalFlatKernels.h
+- vendor/faiss/faiss/gpu_metal/MetalIndex.h
+- vendor/faiss/faiss/gpu_metal/MetalIndexFlat.h
+- vendor/faiss/faiss/gpu_metal/MetalKernels.h
+- vendor/faiss/faiss/gpu_metal/MetalResources.h
+- vendor/faiss/faiss/gpu_metal/StandardMetalResources.h
+- vendor/faiss/faiss/impl/AdSampling.cpp
+- vendor/faiss/faiss/impl/AdSampling.h
 - vendor/faiss/faiss/impl/AdditiveQuantizer.cpp
 - vendor/faiss/faiss/impl/AdditiveQuantizer.h
 - vendor/faiss/faiss/impl/AuxIndexStructures.cpp
 - vendor/faiss/faiss/impl/AuxIndexStructures.h
+- vendor/faiss/faiss/impl/ClusteringHelpers.cpp
+- vendor/faiss/faiss/impl/ClusteringHelpers.h
+- vendor/faiss/faiss/impl/ClusteringInitialization.cpp
+- vendor/faiss/faiss/impl/ClusteringInitialization.h
 - vendor/faiss/faiss/impl/CodePacker.cpp
 - vendor/faiss/faiss/impl/CodePacker.h
+- vendor/faiss/faiss/impl/CodePackerRaBitQ.cpp
+- vendor/faiss/faiss/impl/CodePackerRaBitQ.h
 - vendor/faiss/faiss/impl/DistanceComputer.h
 - vendor/faiss/faiss/impl/FaissAssert.h
 - vendor/faiss/faiss/impl/FaissException.cpp
 - vendor/faiss/faiss/impl/FaissException.h
-- vendor/faiss/faiss/impl/FastScanDistancePostProcessing.h
 - vendor/faiss/faiss/impl/HNSW.cpp
 - vendor/faiss/faiss/impl/HNSW.h
 - vendor/faiss/faiss/impl/IDSelector.cpp
 - vendor/faiss/faiss/impl/IDSelector.h
+- vendor/faiss/faiss/impl/InvertedListScannerStats.h
 - vendor/faiss/faiss/impl/LocalSearchQuantizer.cpp
 - vendor/faiss/faiss/impl/LocalSearchQuantizer.h
-- vendor/faiss/faiss/impl/LookupTableScaler.h
 - vendor/faiss/faiss/impl/NNDescent.cpp
 - vendor/faiss/faiss/impl/NNDescent.h
 - vendor/faiss/faiss/impl/NSG.cpp
@@ -253,6 +272,8 @@ files:
 - vendor/faiss/faiss/impl/Panorama.h
 - vendor/faiss/faiss/impl/PanoramaStats.cpp
 - vendor/faiss/faiss/impl/PanoramaStats.h
+- vendor/faiss/faiss/impl/PdxLayout.cpp
+- vendor/faiss/faiss/impl/PdxLayout.h
 - vendor/faiss/faiss/impl/PolysemousTraining.cpp
 - vendor/faiss/faiss/impl/PolysemousTraining.h
 - vendor/faiss/faiss/impl/ProductAdditiveQuantizer.cpp
@@ -261,8 +282,6 @@ files:
 - vendor/faiss/faiss/impl/ProductQuantizer.cpp
 - vendor/faiss/faiss/impl/ProductQuantizer.h
 - vendor/faiss/faiss/impl/Quantizer.h
-- vendor/faiss/faiss/impl/RaBitQStats.cpp
-- vendor/faiss/faiss/impl/RaBitQStats.h
 - vendor/faiss/faiss/impl/RaBitQUtils.cpp
 - vendor/faiss/faiss/impl/RaBitQUtils.h
 - vendor/faiss/faiss/impl/RaBitQuantizer.cpp
@@ -276,11 +295,49 @@ files:
 - vendor/faiss/faiss/impl/ScalarQuantizer.h
 - 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-sve.h
-- vendor/faiss/faiss/impl/code_distance/code_distance.h
+- vendor/faiss/faiss/impl/VisitedTable.cpp
+- vendor/faiss/faiss/impl/VisitedTable.h
+- vendor/faiss/faiss/impl/approx_topk/approx_topk.h
+- vendor/faiss/faiss/impl/approx_topk/avx2.cpp
+- vendor/faiss/faiss/impl/approx_topk/generic.h
+- vendor/faiss/faiss/impl/approx_topk/neon.cpp
+- vendor/faiss/faiss/impl/approx_topk/rq_beam_search_tab-inl.h
+- vendor/faiss/faiss/impl/approx_topk/rq_beam_search_tab.h
+- vendor/faiss/faiss/impl/approx_topk/simdlib256-inl.h
+- vendor/faiss/faiss/impl/binary_hamming/IndexBinaryHNSW_impl.h
+- vendor/faiss/faiss/impl/binary_hamming/IndexBinaryHash_impl.h
+- vendor/faiss/faiss/impl/binary_hamming/IndexBinaryIVF_impl.h
+- vendor/faiss/faiss/impl/binary_hamming/IndexIVFSpectralHash_impl.h
+- vendor/faiss/faiss/impl/binary_hamming/IndexPQ_impl.h
+- vendor/faiss/faiss/impl/binary_hamming/avx2.cpp
+- vendor/faiss/faiss/impl/binary_hamming/avx512.cpp
+- vendor/faiss/faiss/impl/binary_hamming/dispatch.h
+- vendor/faiss/faiss/impl/binary_hamming/neon.cpp
+- vendor/faiss/faiss/impl/binary_hamming/rvv.cpp
+- vendor/faiss/faiss/impl/expanded_scanners.h
+- vendor/faiss/faiss/impl/fast_scan/FastScanDistancePostProcessing.h
+- vendor/faiss/faiss/impl/fast_scan/LookupTableScaler.h
+- vendor/faiss/faiss/impl/fast_scan/accumulate_loops.h
+- vendor/faiss/faiss/impl/fast_scan/accumulate_loops_512.h
+- vendor/faiss/faiss/impl/fast_scan/decompose_qbs.h
+- vendor/faiss/faiss/impl/fast_scan/dispatching.h
+- vendor/faiss/faiss/impl/fast_scan/fast_scan.cpp
+- vendor/faiss/faiss/impl/fast_scan/fast_scan.h
+- vendor/faiss/faiss/impl/fast_scan/impl-avx2.cpp
+- vendor/faiss/faiss/impl/fast_scan/impl-avx512.cpp
+- vendor/faiss/faiss/impl/fast_scan/impl-neon.cpp
+- vendor/faiss/faiss/impl/fast_scan/impl-riscv.cpp
+- vendor/faiss/faiss/impl/fast_scan/kernels_simd256.h
+- vendor/faiss/faiss/impl/fast_scan/kernels_simd512.h
+- vendor/faiss/faiss/impl/fast_scan/rabitq_dispatching.h
+- vendor/faiss/faiss/impl/fast_scan/rabitq_result_handler.h
+- vendor/faiss/faiss/impl/fast_scan/simd_result_handlers.h
+- vendor/faiss/faiss/impl/hnsw/LockVector.cpp
+- vendor/faiss/faiss/impl/hnsw/LockVector.h
+- vendor/faiss/faiss/impl/hnsw/MinimaxHeap.cpp
+- vendor/faiss/faiss/impl/hnsw/MinimaxHeap.h
+- vendor/faiss/faiss/impl/hnsw/avx2.cpp
+- vendor/faiss/faiss/impl/hnsw/avx512.cpp
 - vendor/faiss/faiss/impl/index_read.cpp
 - vendor/faiss/faiss/impl/index_read_utils.h
 - vendor/faiss/faiss/impl/index_write.cpp
@@ -295,13 +352,42 @@ files:
 - vendor/faiss/faiss/impl/mapped_io.h
 - vendor/faiss/faiss/impl/maybe_owned_vector.h
 - vendor/faiss/faiss/impl/platform_macros.h
-- vendor/faiss/faiss/impl/pq4_fast_scan.cpp
-- 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/pq_code_distance/IVFPQScanner_impl.h
+- vendor/faiss/faiss/impl/pq_code_distance/IVFPQ_QueryTables.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/IVFPQ_QueryTables.h
+- vendor/faiss/faiss/impl/pq_code_distance/PQDistanceComputer_impl.h
+- vendor/faiss/faiss/impl/pq_code_distance/avx2.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/avx512.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/neon.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-avx2.h
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-avx512.h
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-generic.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-generic.h
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-inl.h
+- vendor/faiss/faiss/impl/pq_code_distance/pq_code_distance-sve.cpp
+- vendor/faiss/faiss/impl/pq_code_distance/rvv.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/impl/scalar_quantizer/codecs.h
+- vendor/faiss/faiss/impl/scalar_quantizer/distance_computers.h
+- vendor/faiss/faiss/impl/scalar_quantizer/quantizers.h
+- vendor/faiss/faiss/impl/scalar_quantizer/scanners.h
+- vendor/faiss/faiss/impl/scalar_quantizer/similarities.h
+- vendor/faiss/faiss/impl/scalar_quantizer/sq-avx2.cpp
+- vendor/faiss/faiss/impl/scalar_quantizer/sq-avx512.cpp
+- vendor/faiss/faiss/impl/scalar_quantizer/sq-dispatch.h
+- vendor/faiss/faiss/impl/scalar_quantizer/sq-neon.cpp
+- vendor/faiss/faiss/impl/scalar_quantizer/sq-rvv.cpp
+- vendor/faiss/faiss/impl/scalar_quantizer/training.cpp
+- vendor/faiss/faiss/impl/scalar_quantizer/training.h
+- vendor/faiss/faiss/impl/simd_dispatch.h
+- vendor/faiss/faiss/impl/simdlib/simdlib.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_avx2.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_avx512.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_dispatch.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_emulated.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_neon.h
+- vendor/faiss/faiss/impl/simdlib/simdlib_ppc64.h
 - vendor/faiss/faiss/impl/svs_io.cpp
 - vendor/faiss/faiss/impl/svs_io.h
 - vendor/faiss/faiss/impl/zerocopy_io.cpp
@@ -324,6 +410,12 @@ files:
 - vendor/faiss/faiss/svs/IndexSVSFaissUtils.h
 - vendor/faiss/faiss/svs/IndexSVSFlat.cpp
 - vendor/faiss/faiss/svs/IndexSVSFlat.h
+- vendor/faiss/faiss/svs/IndexSVSIVF.cpp
+- vendor/faiss/faiss/svs/IndexSVSIVF.h
+- vendor/faiss/faiss/svs/IndexSVSIVFLVQ.cpp
+- vendor/faiss/faiss/svs/IndexSVSIVFLVQ.h
+- vendor/faiss/faiss/svs/IndexSVSIVFLeanVec.cpp
+- vendor/faiss/faiss/svs/IndexSVSIVFLeanVec.h
 - vendor/faiss/faiss/svs/IndexSVSVamana.cpp
 - vendor/faiss/faiss/svs/IndexSVSVamana.h
 - vendor/faiss/faiss/svs/IndexSVSVamanaLVQ.cpp
@@ -337,22 +429,18 @@ files:
 - 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
-- vendor/faiss/faiss/utils/approx_topk/avx2-inl.h
-- 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_dispatch.h
 - vendor/faiss/faiss/utils/distances_fused/avx512.cpp
-- vendor/faiss/faiss/utils/distances_fused/avx512.h
 - vendor/faiss/faiss/utils/distances_fused/distances_fused.cpp
 - vendor/faiss/faiss/utils/distances_fused/distances_fused.h
 - vendor/faiss/faiss/utils/distances_fused/simdlib_based.cpp
-- vendor/faiss/faiss/utils/distances_fused/simdlib_based.h
+- vendor/faiss/faiss/utils/distances_fused/simdlib_based_neon.cpp
+- vendor/faiss/faiss/utils/distances_fused/simdlib_kernel-inl.h
 - vendor/faiss/faiss/utils/distances_simd.cpp
-- 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
@@ -362,27 +450,53 @@ files:
 - vendor/faiss/faiss/utils/hamming-inl.h
 - vendor/faiss/faiss/utils/hamming.cpp
 - vendor/faiss/faiss/utils/hamming.h
-- vendor/faiss/faiss/utils/hamming_distance/avx2-inl.h
-- vendor/faiss/faiss/utils/hamming_distance/avx512-inl.h
 - vendor/faiss/faiss/utils/hamming_distance/common.h
-- vendor/faiss/faiss/utils/hamming_distance/generic-inl.h
-- vendor/faiss/faiss/utils/hamming_distance/hamdis-inl.h
-- vendor/faiss/faiss/utils/hamming_distance/neon-inl.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_avx2.cpp
+- vendor/faiss/faiss/utils/hamming_distance/hamming_avx512.cpp
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer-avx2.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer-avx512.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer-generic.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer-neon.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer-rvv.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_computer.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_impl.h
+- vendor/faiss/faiss/utils/hamming_distance/hamming_neon.cpp
+- vendor/faiss/faiss/utils/hamming_distance/hamming_rvv.cpp
 - vendor/faiss/faiss/utils/ordered_key_value.h
 - vendor/faiss/faiss/utils/partitioning.cpp
 - vendor/faiss/faiss/utils/partitioning.h
+- vendor/faiss/faiss/utils/popcount.h
+- vendor/faiss/faiss/utils/pq_code_distance.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/rabitq_simd.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/simd_impl/IVFFlatScanner-inl.h
+- vendor/faiss/faiss/utils/simd_impl/distances_aarch64.cpp
+- vendor/faiss/faiss/utils/simd_impl/distances_arm_sve.cpp
+- vendor/faiss/faiss/utils/simd_impl/distances_autovec-inl.h
+- vendor/faiss/faiss/utils/simd_impl/distances_avx2.cpp
+- vendor/faiss/faiss/utils/simd_impl/distances_avx512.cpp
+- vendor/faiss/faiss/utils/simd_impl/distances_rvv.cpp
+- vendor/faiss/faiss/utils/simd_impl/distances_simdlib256.h
+- vendor/faiss/faiss/utils/simd_impl/distances_sse-inl.h
+- vendor/faiss/faiss/utils/simd_impl/exhaustive_L2sqr_blas_cmax.h
+- vendor/faiss/faiss/utils/simd_impl/hamming_impl.h
+- vendor/faiss/faiss/utils/simd_impl/partitioning_avx2.cpp
+- vendor/faiss/faiss/utils/simd_impl/partitioning_neon.cpp
+- vendor/faiss/faiss/utils/simd_impl/partitioning_simdlib256.h
+- vendor/faiss/faiss/utils/simd_impl/rabitq_avx2.cpp
+- vendor/faiss/faiss/utils/simd_impl/rabitq_avx512.cpp
+- vendor/faiss/faiss/utils/simd_impl/rabitq_neon.cpp
+- vendor/faiss/faiss/utils/simd_impl/rabitq_rvv.cpp
+- vendor/faiss/faiss/utils/simd_impl/super_kmeans_dispatch.h
+- vendor/faiss/faiss/utils/simd_impl/super_kmeans_kernels.h
+- vendor/faiss/faiss/utils/simd_impl/super_kmeans_kernels_avx2.cpp
+- vendor/faiss/faiss/utils/simd_impl/super_kmeans_kernels_avx512.cpp
+- vendor/faiss/faiss/utils/simd_levels.cpp
+- vendor/faiss/faiss/utils/simd_levels.h
 - vendor/faiss/faiss/utils/sorting.cpp
 - vendor/faiss/faiss/utils/sorting.h
 - vendor/faiss/faiss/utils/transpose/transpose-avx2-inl.h
@@ -400,14 +514,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.2'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.3
+rubygems_version: 4.0.10
 specification_version: 4
 summary: Efficient similarity search and clustering for Ruby
 test_files: []

data/vendor/faiss/faiss/impl/RaBitQStats.cpp

@@ -1,29 +0,0 @@
-/*
- * Copyright (c) Meta Platforms, Inc. and affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
-
-#include <faiss/impl/RaBitQStats.h>
-
-namespace faiss {
-
-// NOLINTNEXTLINE(facebook-avoid-non-const-global-variables)
-RaBitQStats rabitq_stats;
-
-void RaBitQStats::reset() {
-    n_1bit_evaluations = 0;
-    n_multibit_evaluations = 0;
-}
-
-double RaBitQStats::skip_percentage() const {
-    const size_t copy_n_1bit_evaluations = n_1bit_evaluations;
-    const size_t copy_n_multibit_evaluations = n_multibit_evaluations;
-    return copy_n_1bit_evaluations > 0
-            ? 100.0 * (copy_n_1bit_evaluations - copy_n_multibit_evaluations) /
-                    copy_n_1bit_evaluations
-            : 0.0;
-}
-
-} // namespace faiss

data/vendor/faiss/faiss/impl/RaBitQStats.h

@@ -1,56 +0,0 @@
-/*
- * Copyright (c) Meta Platforms, Inc. and affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
-
-#pragma once
-
-#include <faiss/impl/platform_macros.h>
-#include <cstddef>
-
-namespace faiss {
-
-/// Statistics for RaBitQ multi-bit two-stage search.
-///
-/// These stats are ONLY collected for multi-bit mode (nb_bits > 1).
-/// In 1-bit mode, there is no two-stage filtering - all candidates are
-/// evaluated with a single distance computation, so there is nothing
-/// meaningful to track. For 1-bit mode, both counters remain 0.
-///
-/// Multi-bit mode uses a two-stage search:
-///   Stage 1: Compute 1-bit lower bound distance for all candidates
-///   Stage 2: Compute full multi-bit distance only for promising candidates
-///
-/// The skip_percentage() metric measures filtering effectiveness:
-/// how many candidates were filtered out by the 1-bit lower bound
-/// without needing the more expensive multi-bit distance computation.
-///
-/// WARNING: Statistics are not robust to internal threading nor to
-/// concurrent RaBitQ searches. Use these values in a single-threaded
-/// context to accurately gauge RaBitQ's filtering effectiveness.
-/// Call reset() before search, then read stats after search completes.
-struct RaBitQStats {
-    /// Number of candidates evaluated using 1-bit (lower bound) distance.
-    /// This is the first stage of two-stage search in multi-bit mode.
-    /// Always 0 in 1-bit mode (stats not tracked).
-    size_t n_1bit_evaluations = 0;
-
-    /// Number of candidates that passed 1-bit filtering and required
-    /// full multi-bit distance computation (second stage).
-    /// Always 0 in 1-bit mode (stats not tracked).
-    size_t n_multibit_evaluations = 0;
-
-    void reset();
-
-    /// Compute percentage of candidates skipped (filtered out by 1-bit stage).
-    /// Returns 0 if no candidates were evaluated (including 1-bit mode).
-    double skip_percentage() const;
-};
-
-/// Global stats for RaBitQ indexes
-// NOLINTNEXTLINE(facebook-avoid-non-const-global-variables)
-FAISS_API extern RaBitQStats rabitq_stats;
-
-} // namespace faiss

data/vendor/faiss/faiss/impl/code_distance/code_distance-generic.h

@@ -1,81 +0,0 @@
-/*
- * Copyright (c) Meta Platforms, Inc. and affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
-
-#pragma once
-
-#include <cstddef>
-#include <cstdint>
-
-namespace faiss {
-
-/// Returns the distance to a single code.
-template <typename PQDecoderT>
-inline float distance_single_code_generic(
-        // number of subquantizers
-        const size_t M,
-        // number of bits per quantization index
-        const size_t nbits,
-        // precomputed distances, layout (M, ksub)
-        const float* sim_table,
-        // the code
-        const uint8_t* code) {
-    PQDecoderT decoder(code, nbits);
-    const size_t ksub = 1 << nbits;
-
-    const float* tab = sim_table;
-    float result = 0;
-
-    for (size_t m = 0; m < M; m++) {
-        result += tab[decoder.decode()];
-        tab += ksub;
-    }
-
-    return result;
-}
-
-/// Combines 4 operations of distance_single_code()
-/// General-purpose version.
-template <typename PQDecoderT>
-inline void distance_four_codes_generic(
-        // number of subquantizers
-        const size_t M,
-        // number of bits per quantization index
-        const size_t nbits,
-        // precomputed distances, layout (M, ksub)
-        const float* sim_table,
-        // codes
-        const uint8_t* __restrict code0,
-        const uint8_t* __restrict code1,
-        const uint8_t* __restrict code2,
-        const uint8_t* __restrict code3,
-        // computed distances
-        float& result0,
-        float& result1,
-        float& result2,
-        float& result3) {
-    PQDecoderT decoder0(code0, nbits);
-    PQDecoderT decoder1(code1, nbits);
-    PQDecoderT decoder2(code2, nbits);
-    PQDecoderT decoder3(code3, nbits);
-    const size_t ksub = 1 << nbits;
-
-    const float* tab = sim_table;
-    result0 = 0;
-    result1 = 0;
-    result2 = 0;
-    result3 = 0;
-
-    for (size_t m = 0; m < M; m++) {
-        result0 += tab[decoder0.decode()];
-        result1 += tab[decoder1.decode()];
-        result2 += tab[decoder2.decode()];
-        result3 += tab[decoder3.decode()];
-        tab += ksub;
-    }
-}
-
-} // namespace faiss