-
Notifications
You must be signed in to change notification settings - Fork 322
[libcu++] Make resource an environment when queried for a resource #7113
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
Conversation
|
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. |
|
/ok to test 6708395 |
This comment has been minimized.
This comment has been minimized.
| //! @brief `__get_memory_resource_t` is a customization point object that queries a type `T` for an associated memory | ||
| //! @brief `get_memory_resource_t` is a customization point object that queries a type `T` for an associated memory | ||
| //! resource | ||
| struct __get_memory_resource_t |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
My memory is fuzzy here, why did we keep this internal last release?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I am not sure, but I think we can just remove the internal name usage
😬 CI Workflow Results🟥 Finished in 5h 43m: Pass: 97%/136 | Total: 5d 14h | Max: 5h 43m | Hits: 88%/272253See results here. |
|
Closing in favor of #7266 |
We introduced environment based APIs in CUB, but right now you need to use:
It would be better to just do
like we do with a stream. This PR adds an overload accepting a resource to
cuda::mr::get_memory_resource, like we do withget_stream. It would be a burden to require every resource to declarecuda::mr::get_memory_resourcequery, so instead this PR changescuda::std::execution::__query_orto also try to call the supplied query with the env as an argument, to utilize the newly added overload incuda::mr::get_memory_resource