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.

Sign up for GitHub

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

Jump to bottom

internal utility for pushing/popping the cuda driver's "current" device #2040

Closed
Tracked by #2039
pciolkosz opened this issue Jul 23, 2024 · 2 comments
Closed
Tracked by #2039

internal utility for pushing/popping the cuda driver's "current" device #2040

pciolkosz opened this issue Jul 23, 2024 · 2 comments
Assignees
@pciolkosz
Labels
CUDA Next Feature intended for the Cuda Next experimental library feature request New feature or request.

Comments

@pciolkosz
Copy link
Contributor

pciolkosz commented Jul 23, 2024 •
edited
Loading

Provide a RAII utility that sets the current CUDA device/context (green context in the future?) and cleans up when the current scope ends. It should lower to the driver API using wrappers from #2042
@pciolkosz pciolkosz mentioned this issue Jul 23, 2024
Open
6 tasks
@github-project-automation github-project-automation bot moved this to Todo in CCCL Jul 23, 2024
@pciolkosz pciolkosz self-assigned this Jul 23, 2024
@pciolkosz pciolkosz added feature request New feature or request. CUDA Next Feature intended for the Cuda Next experimental library labels Jul 23, 2024
@ericniebler ericniebler changed the title internal utility for pushing/pupping the cuda driver's "current" device internal utility for pushing/popping the cuda driver's "current" device Jul 23, 2024
@bernhardmgruber
Copy link
Contributor

This sounds similar to cub::SwitchDevice. Maybe your new facility can replace/build upon it.

@pciolkosz
Copy link
Contributor Author

Ideally, once the C++ runtime is finished you won't need it anymore 😄
On more serious note, because we want to use driver API here, it might not be possible to unify it with the one from CUB that is using the runtime, but we can try.

@bernhardmgruber bernhardmgruber mentioned this issue Jul 25, 2024
Merged
This was referenced Jul 25, 2024
Merged
Merged
@github-project-automation github-project-automation bot moved this from Todo to Done in CCCL Jul 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@pciolkosz pciolkosz

Labels
CUDA Next Feature intended for the Cuda Next experimental library feature request New feature or request.
Projects
CCCL
Archived in project
Milestone
No milestone
Development

No branches or pull requests
3 participants
@ericniebler @bernhardmgruber @pciolkosz