Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA] Improvements to RMM stream pool #1340

Open
harrism opened this issue Sep 13, 2023 · 0 comments
Open

[FEA] Improvements to RMM stream pool #1340

harrism opened this issue Sep 13, 2023 · 0 comments
Labels
0 - Backlog In queue waiting for assignment feature request New feature or request

Comments

@harrism
Copy link
Member

harrism commented Sep 13, 2023

Is your feature request related to a problem? Please describe.
Extend RMM stream_pool with the same features as were recently added to libcudf in rapidsai/cudf#13922

  • add a base stream_pool interface class
  • implement the interface using a pool like the current one
  • Also implement a default_stream_pool that always returns the default stream (for debugging)
  • Add fork / join functions for forking a set of streams from a specified stream (using the pool to supply streams), and to join a list of streams to another stream.

Describe the solution you'd like
rapidsai/cudf#13922

@harrism harrism added feature request New feature or request 0 - Backlog In queue waiting for assignment labels Sep 13, 2023
@harrism harrism self-assigned this Oct 10, 2023
@harrism harrism removed their assignment Jan 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
0 - Backlog In queue waiting for assignment feature request New feature or request
Projects
Status: Todo
Development

No branches or pull requests

1 participant