rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.26k stars 535 forks source link

[FEA] Use RAFT get/set_current_device_resource() wrappers in cuML #6058

Open harrism opened 2 months ago

harrism commented 2 months ago

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.