Skip to content

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

Merged
rapids-bot[bot] merged 1 commit into
NVIDIA:mainfrom
viclafargue:harden-mg-ann-tests
Jul 1, 2026
Merged

Harden multi-GPU ANN tests for larger GPU counts#2280
rapids-bot[bot] merged 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.

@cjnolet

cjnolet commented Jul 1, 2026

Copy link
Copy Markdown
Contributor

/merge

@rapids-bot rapids-bot Bot merged commit 990d227 into NVIDIA:main Jul 1, 2026
73 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

Status: In Progress

Development

Successfully merging this pull request may close these issues.

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

3 participants