cuvs
cuvs copied to clipboard
[FEA] Use RAFT get/set_current_device_resource() wrappers in cuVS
Is your feature request related to a problem? Please describe.
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.
Using these will allow us to change them in a single location in RAFT without having to touch every file that uses memory resources.
Additional context See https://github.com/rapidsai/raft/issues/2423 and https://github.com/rapidsai/raft/pull/2424