Multi partition cagra search#2035

Draft

jamxia155 wants to merge 31 commits into

rapidsai:mainfrom

jamxia155:multi-segment-cagra-search

Commits on Apr 22, 2026

Add multi-segment GPU search: cuvsSelectK C API and Java bindings
jamxia155
committed
Add parallel multi-segment CAGRA search with async memory resource
jamxia155
committed
Fix CudaStreamPool races and reduce multi-segment search overhead
jamxia155
committed
Default initialize CAGRA search parameters
jamxia155
committed
Enable persistent CAGRA kernel to be shared across index segments
jamxia155
committed
java: parallelize persistent-mode segment searches in MultiSegmentCagraSearch

jamxia155
and
claude
committed
Add native multi-segment CAGRA search and per-resources workspace pool
jamxia155
committed
java: add DataType.HALF (float16) support to CuVSMatrix
jamxia155
committed

Commits on Apr 23, 2026

Commits on May 1, 2026

Remove remnants of experimental code
jamxia155
committed

Commits on May 11, 2026

Load libcudart.so automatically when libcuvs_c.so uses static CUDA
jamxia155
committed

Commits on May 16, 2026

Add multi-segment CAGRA search with bitset prefilter support
jamxia155
committed

Commits on May 18, 2026

Add prefiltered ground truth generation to cuvs_bench
jamxia155
committed

Commits on May 20, 2026

Commits on May 21, 2026

Refactor multi-partition CAGRA search to return merged global top-k
jamxia155
committed

Commits on May 30, 2026

Allow multi-partition CAGRA search topk to exceed per-partition itopk_size
jamxia155
committed

Commits on Jun 1, 2026

Revert "Allow multi-partition CAGRA search topk to exceed per-partition itopk_size"
jamxia155
committed

Commits on Jun 2, 2026

Commits on Jun 3, 2026

Commits on Jun 5, 2026

Commits on Jun 8, 2026

Commits on Jun 9, 2026

Gate multi-partition AUTO on (query, partition) CTA count
jamxia155
committed

Commits on Jun 10, 2026

Merge remote-tracking branch 'origin/main' into multi-segment-cagra-search
jamxia155
committed