cugraph
cugraph copied to clipboard
[FEA]: Use RAFT get/set_current_device_resource() wrappers in cuGraph
Is this a new feature, an improvement, or a change to existing functionality?
New Feature
How would you describe the priority of this feature request
High
Please provide a clear description of problem this feature solves
To reduce thrash due to RMM refactoring, RAFT is adding wrappers around RMM rmm::get_current_device_resource(), rmm::set_current_device_resource() and aliases for RMM resource_ref types.
See rapidsai/raft#2424
Using these will allow us to change them in a single location in RAFT without having to touch every file that uses memory resources.
Describe your ideal solution
See above
Describe any alternatives you have considered
None
Additional context
No response
Code of Conduct
- [x] I agree to follow cuGraph's Code of Conduct
- [x] I have searched the open feature requests and have found no duplicates for this feature request