Fix chunk-based processing in ResidualCoarseQuantizer::search() (#3047)

Summary: Adds a missing function argument to ResidualCoarseQuantizer() whenever the data is processed in chunks Pull Request resolved: #3047 Reviewed By: mlomeli1 Differential Revision: D49687858 Pulled By: mdouze fbshipit-source-id: 1456138fe1ff3a033b73e97f16470ac8ceca60ab
facebookresearch · Sep 28, 2023 · e18de23 · e18de23
1 parent a1814be
commit e18de23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/faiss/IndexAdditiveQuantizer.cpp b/faiss/IndexAdditiveQuantizer.cpp
@@ -560,7 +560,7 @@ void ResidualCoarseQuantizer::search(
         }
         for (idx_t i0 = 0; i0 < n; i0 += bs) {
             idx_t i1 = std::min(n, i0 + bs);
-            search(i1 - i0, x + i0 * d, k, distances + i0 * k, labels + i0 * k);
+            search(i1 - i0, x + i0 * d, k, distances + i0 * k, labels + i0 * k, params_in);
             InterruptCallback::check();
         }
         return;