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] Convert rmm resource parameters to rmm::device_async_resource_ref #1979

Open
ttnghia opened this issue Apr 18, 2024 · 1 comment
Open
Assignees

Comments

@ttnghia
Copy link
Collaborator

ttnghia commented Apr 18, 2024

In order to move along with rmm code refactoring, similar to rapidsai/cudf#15507, we need to perform a repository-wide conversion from rmm::mr::device_memory_resource * into rmm::device_async_resource_ref.

This needs to be done ASAP before rmm::mr::device_memory_resource * is removed from rmm.

@pmattione-nvidia
Copy link
Collaborator

Merged PR 2011 does much of this work; this conversion has now been performed at all of the kernel interfaces. However the resource allocator in SparkResourceAdaptor.jni still uses device_memory_resource and needs to be updated.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

4 participants