Skip to content

Use cuda::stream_ref for central stream helpers - #23770

Draft
vyasr wants to merge 2 commits into
NVIDIA:mainfrom
vyasr:codex/cuda-stream-ref-central-helpers
Draft

Use cuda::stream_ref for central stream helpers#23770
vyasr wants to merge 2 commits into
NVIDIA:mainfrom
vyasr:codex/cuda-stream-ref-central-helpers

Conversation

@vyasr

@vyasr vyasr commented Aug 23, 2026

Copy link
Copy Markdown
Contributor

Description

This Wave 2 follow-up to #23691 migrates the central stream helpers from rmm::cuda_stream_view to cuda::stream_ref, including get_default_stream(), default_stream_value, fork_streams, and join_streams.

It also updates the direct Java, pylibcudf, and libcudf_streaming call sites whose method calls depend on those helper return types. rmm::cuda_stream ownership remains unchanged.

This should land after the Wave 1 downstream stream_ref PRs.

Replaces the fork PR vyasr#9.

Contributes to #23636

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@copy-pr-bot

copy-pr-bot Bot commented Aug 23, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@vyasr vyasr added libcudf Affects libcudf (C++/CUDA) code. Python Affects Python cuDF API. Java Affects Java cuDF API. pylibcudf Issues specific to the pylibcudf package improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Aug 23, 2026
@github-actions github-actions Bot removed improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Aug 23, 2026
@vyasr vyasr added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Aug 23, 2026
stream_refs.reserve(streams.size());
for (auto const s : streams) {
stream_refs.emplace_back(s);
}

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Silly question: Why is this not std::copy() to back_inserter{stream_refs}?

@mythrocks mythrocks left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yep. 👍

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

Labels

improvement Improvement / enhancement to an existing function Java Affects Java cuDF API. libcudf Affects libcudf (C++/CUDA) code. non-breaking Non-breaking change pylibcudf Issues specific to the pylibcudf package Python Affects Python cuDF API.

Projects

Status: Todo

Development

Successfully merging this pull request may close these issues.

2 participants