Optimize transfer/compute overlap in out-of-core KMeans - #2538

Merged

rapids-bot[bot] merged 18 commits into

NVIDIA:release/26.10from

viclafargue:ooc-kmeans-overlap

Sep 18, 2026

Commits on Sep 15, 2026

Optimize out-of-core KMeans overlap
viclafargue
committed
d3e4ab9
Refine out-of-core KMeans batch pipelining
viclafargue
committed
8c88dc7
multi-GPU transfer/compute overlap
viclafargue
committed
2a00053
Respond to feedback
viclafargue
committed
d59c6a1
Allow users to set a stream pool from Python
viclafargue
committed
45d640c
address feedback
viclafargue
committed
8e50ae9
call RAFT set_stream_pool
viclafargue
committed
4025b8e

Commits on Sep 16, 2026

Improve final inertia transfer/compute overlap
viclafargue
committed
f53723e
Check style
viclafargue
committed
5928c87

Commits on Sep 17, 2026

Remove cluster_cost_workspace struct
viclafargue
committed
391911d
Proper documentation of OOC KMeans
viclafargue
committed
ffb4a73
Removing C/Python API
viclafargue
committed
c83068c
Use get_prefetch_stream
viclafargue
committed
c9703b9
Revert get_hnswlib.cmake
viclafargue
committed
83cd98a

Commits on Sep 18, 2026

large workspace + kmeans_batch_loader refactor
viclafargue
committed
4fd519c
More refactor
viclafargue
committed
782da16
remove doc updates
viclafargue
committed
6357898
Merge branch 'release/26.10' into ooc-kmeans-overlap
viclafargue
committed
bfea61c