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] Draw a worker stream from a parent handle and set it as the user stream​ in a returned handle #142

Closed
afender opened this issue Feb 8, 2021 · 0 comments · Fixed by #148
Assignees
Labels
feature request New feature or request

Comments

@afender
Copy link
Member

afender commented Feb 8, 2021

Is your feature request related to a problem? Please describe.
Many functions (including all graph prims) accept a handle and execute on the handle's user stream (accessed by handle_t.get_stream()​). This is an issue if the caller is also provided with a single handle but wants to run these functions concurrently. We need to leverage the pool of worker streams of the parent handle without having to change the signature and internals of low-level functions.

Describe the solution you'd like
One way is to add a raft::handle_t get_handle_from_pool feature that draws a worker stream from the parent handle and set it as the user stream​ of the returned handle.

Describe alternatives you've considered
Pass streams explicitly or rewrite prims to access worker streams from the handle

Additional context
Needed for rapidsai/cugraph#957

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant