Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
harrism opened this issue Sep 4, 2024 · 0 comments
Open

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

harrism opened this issue Sep 4, 2024 · 0 comments
Assignees
Labels
? - Needs Triage Need team to review and classify feature request New feature or request

Comments

@harrism
Copy link
Member

harrism commented Sep 4, 2024

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.

@harrism harrism added ? - Needs Triage Need team to review and classify feature request New feature or request labels Sep 4, 2024
@harrism harrism self-assigned this Sep 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
? - Needs Triage Need team to review and classify feature request New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant