内存资源管理#
- group Memory Resource Management
函数
-
inline rmm::mr::device_memory_resource *get_current_device_resource()#
获取当前设备内存资源。
- 返回：
当前设备内存资源。
-
inline rmm::device_async_resource_ref get_current_device_resource_ref()#
获取当前设备内存资源引用。
- 返回：
当前设备内存资源引用。
-
inline rmm::mr::device_memory_resource *set_current_device_resource(rmm::mr::device_memory_resource *mr)#
设置当前设备内存资源。
- 参数：
mr – 新的设备内存资源。
- 返回：
先前的设备内存资源。
-
inline rmm::device_async_resource_ref set_current_device_resource_ref(rmm::device_async_resource_ref mr)#
设置当前设备内存资源引用。
- 参数：
**mr** – 新的设备内存资源引用。
- 返回：
先前的设备内存资源引用。
-
inline rmm::device_async_resource_ref reset_current_device_resource_ref()#
将当前设备内存资源引用重置为初始资源。
- 返回：
先前的设备内存资源引用。
-
inline rmm::mr::device_memory_resource *get_current_device_resource()#