faiss 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '02752578b25cfcc61a7bb0e06e1d0dbc2d78175796cf99a44184a4e7298de073'
-  data.tar.gz: ec678cf766efa90be0e013b342be6f95e42c6e67c4aa6de1b9e3b6f4c59d8699
+  metadata.gz: f6fb14299c0b91d6ddaf74ca2fa2d6895e8621249a5351683b109f35866ab541
+  data.tar.gz: 2ec9d11faf8f3f1af7790d1f39e31cb92d8a79d2539fa7c935dbda63f3052d8c
 SHA512:
-  metadata.gz: 9342c736f6ce775b8561f0fc066b565e38f8d38470f98ea51b9635f371bb8efdbb9e049874588847fb11360da006da3ec0ca28d44fc6f844d970d91f648f1b03
-  data.tar.gz: e70f220fdc76a37fcf5a06beba2b07f0315bbbb3dab7b396a8f932c04746e11454bd57ed527d18d55436352bd823d1e33e341046697fd36b124f7e64fbb72f38
+  metadata.gz: 5afa3a7ca6947fad6f5a4f7693b524411875d212079621bc72bda73df36fdf090fcb566b83248302a83ecb38ee04189c57c812eaad4ffc99f94840eaf3abfd28
+  data.tar.gz: c29cf727d683f82e530ba1788630ae546ddb4470f3145f0a4dd1b6afe31555bb9363dfa52cc2a5e005a077f6e2aae24b3dc1aa44d1ff9c50fc9269987df0ec87

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.7 (2021-03-26)
+
+- Added `IndexScalarQuantizer` and `IndexIVFScalarQuantizer`
+
 ## 0.1.6 (2021-03-21)
 
 - Added support for inner product to `IndexIVFFlat` and `IndexIVFPQ`

data/README.md

@@ -97,12 +97,24 @@ Locality-sensitive hashing
 Faiss::IndexLSH.new(d, nbits)
 ```
 
+Scalar quantizer (SQ) in flat mode
+
+```rb
+Faiss::IndexScalarQuantizer.new(d, qtype)
+```
+
 Product quantizer (PQ) in flat mode
 
 ```rb
 Faiss::IndexPQ.new(d, m, nbits)
 ```
 
+IVF and scalar quantizer
+
+```rb
+Faiss::IndexIVFScalarQuantizer.new(quantizer, d, nlists, qtype)
+```
+
 IVFADC (coarse quantizer+PQ on residuals)
 
 ```rb

data/ext/faiss/index.cpp

@@ -30,6 +30,30 @@ faiss::MetricType from_ruby<faiss::MetricType>(Rice::Object x)
   }
 }
 
+template<>
+faiss::ScalarQuantizer::QuantizerType from_ruby<faiss::ScalarQuantizer::QuantizerType>(Rice::Object x)
+{
+  auto s = x.to_s().str();
+  if (s == "qt_8bit") {
+    return faiss::ScalarQuantizer::QT_8bit;
+  } else if (s == "qt_4bit") {
+    return faiss::ScalarQuantizer::QT_4bit;
+  } else if (s == "qt_8bit_uniform") {
+    return faiss::ScalarQuantizer::QT_8bit_uniform;
+  } else if (s == "qt_4bit_uniform") {
+    return faiss::ScalarQuantizer::QT_4bit_uniform;
+  } else if (s == "qt_fp16") {
+    return faiss::ScalarQuantizer::QT_fp16;
+  } else if (s == "qt_8bit_direct") {
+    return faiss::ScalarQuantizer::QT_8bit_direct;
+  } else if (s == "qt_6bit") {
+    return faiss::ScalarQuantizer::QT_6bit;
+  } else {
+    // TODO throw argument error
+    throw std::runtime_error("Invalid quantizer type: " + s);
+  }
+}
+
 void init_index(Rice::Module& m) {
   Rice::define_class_under<faiss::Index>(m, "Index")
     .define_method(

data/lib/faiss/version.rb

@@ -1,3 +1,3 @@
 module Faiss
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: faiss
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-03-21 00:00:00.000000000 Z
+date: 2021-03-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice