Skip to content

Harden multi-GPU ANN tests for larger GPU counts#2280

Open
viclafargue wants to merge 1 commit into
NVIDIA:mainfrom
viclafargue:harden-mg-ann-tests
Open

Harden multi-GPU ANN tests for larger GPU counts#2280
viclafargue wants to merge 1 commit into
NVIDIA:mainfrom
viclafargue:harden-mg-ann-tests

Conversation

@viclafargue

Copy link
Copy Markdown
Contributor

Closes #2096.

Harden MG IVF-Flat, IVF-PQ, and CAGRA tests for larger GPU counts by scaling sharded test data, validating returned IDs, fixing mode-specific search coverage, and using shard-local append IDs for non-empty IVF extends.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[BUG] IVF-FLAT recall significantly worse on multi-GPU Blackwell (B200, GB200)

1 participant