Skip to content

Commit

Permalink
Fix chunk-based processing in ResidualCoarseQuantizer::search() (face…
Browse files Browse the repository at this point in the history
…bookresearch#3047)

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
  • Loading branch information
alexanderguzhva authored and facebook-github-bot committed Sep 28, 2023
1 parent a1814be commit e18de23
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion faiss/IndexAdditiveQuantizer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit e18de23

Please sign in to comment.