mdouze

mdouze

diff --git a/faiss/gpu/test/test_pytorch_faiss.py b/faiss/gpu/test/test_pytorch_faiss.py
--- a/faiss/gpu/test/test_pytorch_faiss.py
+++ b/faiss/gpu/test/test_pytorch_faiss.py
@@ -86,10 +86,19 @@
     D_ptr = swig_ptr_from_FloatTensor(D)
     I_ptr = swig_ptr_from_LongTensor(I)
 
-    faiss.bruteForceKnn(res, metric,
-                        xb_ptr, xb_row_major, nb,
-                        xq_ptr, xq_row_major, nq,

mdouze

mdouze

mdouze

mdouze

mdouze

diff --git a/faiss/index_factory.cpp b/faiss/index_factory.cpp
--- a/faiss/index_factory.cpp
+++ b/faiss/index_factory.cpp
@@ -81,6 +81,7 @@
 
     int64_t ncentroids = -1;
     bool use_2layer = false;
+    int hnsw_M = -1;
 
     for (char *tok = strtok_r (description, " ,", &ptr);

mdouze

mdouze

mdouze