Skip to content

[DO NOT MERGE] Reproduce IVF-RaBitQ wrong-list results for queries on a list centroid - #2800

Closed
bdice wants to merge 3 commits into
NVIDIA:mainfrom
bdice:fix-ivf-rabitq-exact-match-search-repro
Closed

bdice wants to merge 3 commits into
NVIDIA:mainfrom
bdice:fix-ivf-rabitq-exact-match-search-repro

Conversation

@bdice

@bdice bdice commented Oct 10, 2026

Copy link
Copy Markdown
Contributor

Reproduces IVF-RaBitQ returning neighbors from the wrong list when a query coincides with a list centroid (for example, a query equal to a duplicated data row). Found by workstream G (#2791, #2798).

Adds ivf_rabitq_exact_match_test: 32 distinct rows, each repeated 128 times, n_lists = 32, and the queries are the distinct rows. All 10 neighbors must be copies of the query at distance ~0, for every search mode and for bits_per_dim 1 and 3.

CI is reduced to the amd64 CUDA 13.3.0 C++ build and NEIGHBORS_ANN_IVF_RABITQ_TEST on H100 and RTX PRO 6000, repeated.

Part of #2783 (workstream K).

@copy-pr-bot

copy-pr-bot Bot commented Oct 10, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@bdice

bdice commented Oct 10, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test 8ba8a8b

@bdice

bdice commented Oct 10, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test e89081b

@bdice

bdice commented Oct 11, 2026

Copy link
Copy Markdown
Contributor Author

Closing: this was a reproducer/diagnostic PR for workstream K of #2783. The fix is in #2802.

@bdice bdice closed this Oct 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

1 participant