Skip to content

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

@harrism

Description

@harrism

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.

Metadata

Metadata

Assignees

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions