内存资源管理#

group memory_resource

函数

inline rmm::mr::device_memory_resource *get_current_device_resource()#

获取当前设备的内存资源。

Returns:

当前设备的内存资源。

inline rmm::device_async_resource_ref get_current_device_resource_ref()#

获取当前设备的内存资源引用。

Returns:

当前设备内存资源参考。

inline rmm::mr::device_memory_resource *set_current_device_resource(rmm::mr::device_memory_resource *mr)#

设置当前设备的内存资源。

Parameters:

mr – 新设备内存资源。

Returns:

之前的设备内存资源。

inline rmm::device_async_resource_ref set_current_device_resource_ref(rmm::device_async_resource_ref mr)#

设置当前设备内存资源引用。

Parameters:

mr – 新设备内存资源引用。

Returns:

之前的设备内存资源引用。

inline rmm::device_async_resource_ref reset_current_device_resource_ref()#

将当前设备内存资源引用重置为初始资源。

Returns:

之前的设备内存资源引用。