You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
I'm about to undertake yet another change to every function that takes an MR parameter. To prevent doing this too many times in the future, I would like to provide a wrapper in RAFT to centralize calls to rmm::get_current_device_resource_ref and aliases for RMM resource_ref types. hopefully this can reduce the number of files that need to change in the future when this functionality changes (e.g. if it moves out of RMM to CCCL).
Describe the solution you'd like
Either add the following to raft/core/resource/device_memory_resource.hpp, or add a new header, say raft/core/resource/memory_resource.hpp. Please provide suggestions on the best place for this. Contents something like this:
Is your feature request related to a problem? Please describe.
I'm about to undertake yet another change to every function that takes an MR parameter. To prevent doing this too many times in the future, I would like to provide a wrapper in RAFT to centralize calls to rmm::get_current_device_resource_ref and aliases for RMM resource_ref types. hopefully this can reduce the number of files that need to change in the future when this functionality changes (e.g. if it moves out of RMM to CCCL).
Describe the solution you'd like
Either add the following to
raft/core/resource/device_memory_resource.hpp
, or add a new header, sayraft/core/resource/memory_resource.hpp
. Please provide suggestions on the best place for this. Contents something like this:Describe alternatives you've considered
Additional context
rapidsai/rmm#1598
related cudf PR: rapidsai/cudf#16679
The text was updated successfully, but these errors were encountered: