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

alexanderguzhva · facebook-github-bot · commit e18de2303e3e · 2023-09-28T02:45:59.000-07:00
Summary: Adds a missing function argument to ResidualCoarseQuantizer() whenever the data is processed in chunks Pull Request resolved: facebookresearch#3047 Reviewed By: mlomeli1 Differential Revision: D49687858 Pulled By: mdouze fbshipit-source-id: 1456138fe1ff3a033b73e97f16470ac8ceca60ab
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;

Original file line number	Diff line number	Diff line change
`@@ -560,7 +560,7 @@ void ResidualCoarseQuantizer::search(`
`560`	`560`	`}`
`561`	`561`	`for (idx_t i0 = 0; i0 < n; i0 += bs) {`
`562`	`562`	`idx_t i1 = std::min(n, i0 + bs);`
`563`		`- search(i1 - i0, x + i0 * d, k, distances + i0 * k, labels + i0 * k);`
	`563`	`+ search(i1 - i0, x + i0 * d, k, distances + i0 * k, labels + i0 * k, params_in);`
`564`	`564`	`InterruptCallback::check();`
`565`	`565`	`}`
`566`	`566`	`return;`