Repository navigation
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref #318
Copy link
Copy link
Closed
1 / 11 of 1 issue completedClosed
1 / 11 of 1 issue completed
Copy link
Labels
Description
Summary
Track the coordinated migration of stream APIs and call sites from rmm::cuda_stream_view to CCCL's cuda::stream_ref. This propagates cuda::stream_ref through RMM containers and memory resources, RAFT resource and handle APIs, downstream C++ interfaces, Python/Cython bindings, benchmarks, tests, and documentation.
Migrations
- Pass
cuda::stream_refthrough stream pools, resource accessors, conditionals, and downstream APIs without converting tormm::cuda_stream_view - Use
cuda::stream_refconstructions for default/legacy/per-thread streamsrmm::cuda_stream_default➡️cuda::stream_ref{cudaStream_t{cudaStreamDefault}}rmm::cuda_stream_legacy➡️cuda::stream_ref{cudaStreamLegacy}rmm::cuda_stream_per_thread➡️cuda::stream_ref{cudaStreamPerThread}
- Use
.get()when calling an API that requires a rawcudaStream_t, including CUDA runtime, library, CUB, and legacy API boundaries (previouslyrmm::cuda_stream_viewusedvalue()) - Use
.sync()when synchronizing acuda::stream_ref(previouslyrmm::cuda_stream_viewusedsynchronize()) - Update Cython declarations and call sites to pass stream references directly where supported
Preparation pull requests
These independently adopt the get() and sync() compatibility aliases from RMM #2537, reducing the migration PRs to the type and signature changes that require RMM #2372.
- RAFT: Adopt CUDA stream compatibility accessors NVIDIA/raft#3136
- cuVS: Adopt CUDA stream compatibility accessors NVIDIA/cuvs#2557
- cuML: Adopt CUDA stream compatibility accessors NVIDIA/cuml#8564
- cuGraph: Adopt CUDA stream compatibility accessors cugraph#5648
- cuOpt: Adopt CUDA stream compatibility accessors NVIDIA/cuopt#1858
- RapidsMPF: Adopt CUDA stream compatibility accessors rapidsmpf#1199
- cuDF: Adopt CUDA stream compatibility accessors NVIDIA/cudf#23995
Pull requests
- RMM: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref rmm#2372
- UCXX: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref ucxx#742
- RapidsMPF: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref rapidsmpf#1181
- cuDF: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/cudf#23929
- Continues work from [FEA] Migrate from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/cudf#23636
- RAFT: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/raft#3129
- cuVS: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/cuvs#2521
- cuML: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/cuml#8529
- cuGraph: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref cugraph#5639
- cuGraph-GNN: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref cugraph-gnn#529
- cuOpt: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref NVIDIA/cuopt#1828
- nvForest: Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref nvforest#209 (closed; maintainers are removing RMM separately)
Reactions are currently unavailable
Activity
Add stream_ref-compatible cuda_stream_view methods (#2537)
[-]Migrate pooled stream usage to cuda::stream_ref[/-][+]Migrate rmm::cuda_stream_view usage to cuda::stream_ref[/+][-]Migrate rmm::cuda_stream_view usage to cuda::stream_ref[/-][+]Migrate RAPIDS stream APIs from rmm::cuda_stream_view to cuda::stream_ref[/+][-]Migrate RAPIDS stream APIs from rmm::cuda_stream_view to cuda::stream_ref[/-][+]Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref[/+]Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#742
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#529
Adopt CUDA stream compatibility accessors (#1858)
Adopt CUDA stream compatibility accessors (#1199)
[26.10] Backport CUDA stream compatibility accessors (#1204)
Adopt CUDA stream compatibility accessors (#3136)
Adopt CUDA stream compatibility accessors (#2557)
1 remaining item
Adopt CUDA stream compatibility accessors (#23995)
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Adopt CUDA stream compatibility accessors (#5648)
Adopt CUDA stream compatibility accessors (NVIDIA#1858)
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#…
Migrate stream APIs from rmm::cuda_stream_view to cuda::stream_ref (#209
bdice commentedon Sep 14, 2026
I think all of this work is complete! Next steps will be to remove
rmm::cuda_stream_viewfrom RMM in 26.12.