Is this a new feature, an improvement, or a change to existing functionality?
New Feature
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.
Is this a new feature, an improvement, or a change to existing functionality?
New Feature
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.