faiss 0.4.3 → 0.5.1

data/vendor/faiss/faiss/utils/random.cpp

@@ -91,7 +91,7 @@ uint64_t SplitMix64RandomGenerator::next() {
  *  more than 1-100 billion values. */
 
 /* Generate a set of random floating point values such that x[i] in [0,1]
-   multi-threading. For this reason, we rely on re-entreant functions.  */
+   multi-threading. For this reason, we rely on re-entrant functions.  */
 void float_rand(float* x, size_t n, int64_t seed) {
     // only try to parallelize on large enough arrays
     const size_t nblock = n < 1024 ? 1 : 1024;

data/vendor/faiss/faiss/utils/simdlib_avx2.h

@@ -21,7 +21,7 @@ namespace faiss {
  * The objective is to separate the different interpretations of the same
  * registers (as a vector of uint8, uint16 or uint32), to provide printing
  * functions, and to give more readable names to the AVX intrinsics. It does not
- * pretend to be exhausitve, functions are added as needed.
+ * pretend to be exhaustive, functions are added as needed.
  */
 
 /// 256-bit representation without interpretation as a vector

data/vendor/faiss/faiss/utils/simdlib_avx512.h

@@ -23,7 +23,7 @@ namespace faiss {
  * The objective is to separate the different interpretations of the same
  * registers (as a vector of uint8, uint16 or uint32), to provide printing
  * functions, and to give more readable names to the AVX intrinsics. It does not
- * pretend to be exhausitve, functions are added as needed.
+ * pretend to be exhaustive, functions are added as needed.
  */
 
 /// 512-bit representation without interpretation as a vector

data/vendor/faiss/faiss/utils/simdlib_neon.h

@@ -805,8 +805,8 @@ struct simd32uint8 {
         uint8_t bytes[32];
         store(bytes);
         detail::simdlib::bin(
-                const_cast<const char(&)[32]>(
-                        reinterpret_cast<char(&)[32]>(bytes)),
+                const_cast<const char (&)[32]>(
+                        reinterpret_cast<char (&)[32]>(bytes)),
                 bits);
     }
 

data/vendor/faiss/faiss/utils/transpose/transpose-avx512-inl.h

@@ -59,7 +59,7 @@ inline void transpose_16x4(
     const __m512 r1 = _mm512_shuffle_f32x4(i0, i1, _MM_SHUFFLE(3, 1, 3, 1));
     // 32 33 34 35 40 41 42 43 48 49 50 51 56 57 58 59
     const __m512 r2 = _mm512_shuffle_f32x4(i2, i3, _MM_SHUFFLE(2, 0, 2, 0));
-    // 52 53 54 55 60 61 62 63 52 53 54 55 60 61 62 63
+    // 36 37 38 39 44 45 46 47 52 53 54 55 60 61 62 63
     const __m512 r3 = _mm512_shuffle_f32x4(i2, i3, _MM_SHUFFLE(3, 1, 3, 1));
 
     //  0  2  4  6  8 10 12 14 16 18 20 22 24 26 28 30

data/vendor/faiss/faiss/utils/utils.cpp

@@ -102,7 +102,10 @@ int sgemv_(
 namespace faiss {
 
 // this will be set at load time from GPU Faiss
-std::string gpu_compile_options;
+std::string& ref_gpu_compile_options() {
+    static std::string gpu_compile_options;
+    return gpu_compile_options;
+}
 
 std::string get_compile_options() {
     std::string options;
@@ -124,7 +127,11 @@ std::string get_compile_options() {
     options += "GENERIC ";
 #endif
 
-    options += gpu_compile_options;
+#ifdef FAISS_ENABLE_SVS
+    options += "SVS ";
+#endif
+
+    options += ref_gpu_compile_options();
 
     return options;
 }

data/vendor/faiss/faiss/utils/utils.h

@@ -8,7 +8,7 @@
 // -*- c++ -*-
 
 /*
- *  A few utilitary functions for similarity search:
+ *  A few utility functions for similarity search:
  * - optimized exhaustive distance and knn search functions
  * - some functions reimplemented from torch for speed
  */
@@ -64,7 +64,7 @@ void matrix_qr(int m, int n, float* a);
 void ranklist_handle_ties(int k, int64_t* idx, const float* dis);
 
 /** count the number of common elements between v1 and v2
- * algorithm = sorting + bissection to avoid double-counting duplicates
+ * algorithm = sorting + bisection to avoid double-counting duplicates
  */
 size_t ranklist_intersection_size(
         size_t k1,

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: faiss
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.1
 platform: ruby
 authors:
 - Andrew Kane
@@ -105,6 +105,8 @@ files:
 - vendor/faiss/faiss/IndexIVFFastScan.h
 - vendor/faiss/faiss/IndexIVFFlat.cpp
 - vendor/faiss/faiss/IndexIVFFlat.h
+- vendor/faiss/faiss/IndexIVFFlatPanorama.cpp
+- vendor/faiss/faiss/IndexIVFFlatPanorama.h
 - vendor/faiss/faiss/IndexIVFIndependentQuantizer.cpp
 - vendor/faiss/faiss/IndexIVFIndependentQuantizer.h
 - vendor/faiss/faiss/IndexIVFPQ.cpp
@@ -115,6 +117,8 @@ files:
 - vendor/faiss/faiss/IndexIVFPQR.h
 - vendor/faiss/faiss/IndexIVFRaBitQ.cpp
 - vendor/faiss/faiss/IndexIVFRaBitQ.h
+- vendor/faiss/faiss/IndexIVFRaBitQFastScan.cpp
+- vendor/faiss/faiss/IndexIVFRaBitQFastScan.h
 - vendor/faiss/faiss/IndexIVFSpectralHash.cpp
 - vendor/faiss/faiss/IndexIVFSpectralHash.h
 - vendor/faiss/faiss/IndexLSH.cpp
@@ -135,6 +139,8 @@ files:
 - vendor/faiss/faiss/IndexPreTransform.h
 - vendor/faiss/faiss/IndexRaBitQ.cpp
 - vendor/faiss/faiss/IndexRaBitQ.h
+- vendor/faiss/faiss/IndexRaBitQFastScan.cpp
+- vendor/faiss/faiss/IndexRaBitQFastScan.h
 - vendor/faiss/faiss/IndexRefine.cpp
 - vendor/faiss/faiss/IndexRefine.h
 - vendor/faiss/faiss/IndexReplicas.cpp
@@ -231,6 +237,7 @@ files:
 - 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
@@ -242,6 +249,10 @@ files:
 - vendor/faiss/faiss/impl/NNDescent.h
 - vendor/faiss/faiss/impl/NSG.cpp
 - vendor/faiss/faiss/impl/NSG.h
+- vendor/faiss/faiss/impl/Panorama.cpp
+- vendor/faiss/faiss/impl/Panorama.h
+- vendor/faiss/faiss/impl/PanoramaStats.cpp
+- vendor/faiss/faiss/impl/PanoramaStats.h
 - vendor/faiss/faiss/impl/PolysemousTraining.cpp
 - vendor/faiss/faiss/impl/PolysemousTraining.h
 - vendor/faiss/faiss/impl/ProductAdditiveQuantizer.cpp
@@ -250,8 +261,14 @@ 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
 - vendor/faiss/faiss/impl/RaBitQuantizer.h
+- vendor/faiss/faiss/impl/RaBitQuantizerMultiBit.cpp
+- vendor/faiss/faiss/impl/RaBitQuantizerMultiBit.h
 - vendor/faiss/faiss/impl/ResidualQuantizer.cpp
 - vendor/faiss/faiss/impl/ResidualQuantizer.h
 - vendor/faiss/faiss/impl/ResultHandler.h
@@ -285,6 +302,8 @@ files:
 - 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/svs_io.cpp
+- vendor/faiss/faiss/impl/svs_io.h
 - vendor/faiss/faiss/impl/zerocopy_io.cpp
 - vendor/faiss/faiss/impl/zerocopy_io.h
 - vendor/faiss/faiss/index_factory.cpp
@@ -302,6 +321,15 @@ files:
 - vendor/faiss/faiss/invlists/OnDiskInvertedLists.h
 - vendor/faiss/faiss/python/python_callbacks.cpp
 - vendor/faiss/faiss/python/python_callbacks.h
+- vendor/faiss/faiss/svs/IndexSVSFaissUtils.h
+- vendor/faiss/faiss/svs/IndexSVSFlat.cpp
+- vendor/faiss/faiss/svs/IndexSVSFlat.h
+- vendor/faiss/faiss/svs/IndexSVSVamana.cpp
+- vendor/faiss/faiss/svs/IndexSVSVamana.h
+- vendor/faiss/faiss/svs/IndexSVSVamanaLVQ.cpp
+- vendor/faiss/faiss/svs/IndexSVSVamanaLVQ.h
+- vendor/faiss/faiss/svs/IndexSVSVamanaLeanVec.cpp
+- vendor/faiss/faiss/svs/IndexSVSVamanaLeanVec.h
 - vendor/faiss/faiss/utils/AlignedTable.h
 - vendor/faiss/faiss/utils/Heap.cpp
 - vendor/faiss/faiss/utils/Heap.h