Memory Resource Management

group memory_resource

Functions

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

Get the current device memory resource.

Returns:

The current device memory resource.

inline rmm::device_async_resource_ref get_current_device_resource_ref()

Get the current device memory resource reference.

Returns:

The current device memory resource reference.

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

Set the current device memory resource.

Parameters:

mr – The new device memory resource.

Returns:

The previous device memory resource.

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

Set the current device memory resource reference.

Parameters:

mr – The new device memory resource reference.

Returns:

The previous device memory resource reference.

inline rmm::device_async_resource_ref reset_current_device_resource_ref()

Reset the current device memory resource reference to the initial resource.

Returns:

The previous device memory resource reference.