Change `rmm::get_current_device_resource`/`set_current_device_resource` to use rmm::device_async_resource_ref

rapidsai / rmm

RAPIDS Memory Manager

https://docs.rapids.ai/api/rmm/stable/

Apache License 2.0

446 stars 188 forks source link

Change `rmm::get_current_device_resource`/`set_current_device_resource` to use rmm::device_async_resource_ref #1596

Open harrism opened 1 week ago

harrism commented 1 week ago

### Tasks
- [ ] https://github.com/rapidsai/rmm/issues/1597
- [ ] Update tests to use the above
- [ ] Update RAPIDS repos to use resource_ref versions
- [ ] Deprecate `get_current_device_resource` and related functions
- [ ] Remove `get_current_device_resource` and related functions.