cugraph icon indicating copy to clipboard operation
cugraph copied to clipboard

[FEA]: Use RAFT get/set_current_device_resource() wrappers in cuGraph

Open harrism opened this issue 1 year ago • 0 comments

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

harrism avatar Sep 04 '24 03:09 harrism