-
Notifications
You must be signed in to change notification settings - Fork 980
Open
Labels
cudf-polarsIssues specific to cudf-polarsIssues specific to cudf-polarsfeature requestNew feature or requestNew feature or request
Description
Is your feature request related to a problem? Please describe.
As part of cudf-polars' work to use CUDA streams, we need a way to join multiple CUDA streams (in, e.g., a join
operation) to avoid race conditions.
Describe the solution you'd like
cudf (C++) offers a join_streams
method:
cudf/cpp/include/cudf/detail/utilities/stream_pool.hpp
Lines 126 to 132 in 144b74b
/** | |
* @brief Synchronize a stream to an event on a set of streams. | |
* | |
* @param streams Streams to wait on. | |
* @param stream Joined stream that synchronizes with the waited-on streams. | |
*/ | |
void join_streams(host_span<rmm::cuda_stream_view const> streams, rmm::cuda_stream_view stream); |
Describe alternatives you've considered
Reimplementing the logic in cudf-polars.
Additional context
Metadata
Metadata
Assignees
Labels
cudf-polarsIssues specific to cudf-polarsIssues specific to cudf-polarsfeature requestNew feature or requestNew feature or request
Type
Projects
Status
In Progress