#include <cudf/utilities/memory_resource.hpp>
#include <rmm/cuda_device.hpp>
#include <rmm/mr/device/device_memory_resource.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
Go to the source code of this file.
Namespaces | |
cudf | |
cuDF interfaces | |
Functions | |
rmm::mr::device_memory_resource * | cudf::get_current_device_resource () |
Get the current device memory resource. More... | |
rmm::device_async_resource_ref | cudf::get_current_device_resource_ref () |
Get the current device memory resource reference. More... | |
rmm::mr::device_memory_resource * | cudf::set_current_device_resource (rmm::mr::device_memory_resource *mr) |
Set the current device memory resource. More... | |
rmm::device_async_resource_ref | cudf::set_current_device_resource_ref (rmm::device_async_resource_ref mr) |
Set the current device memory resource reference. More... | |
rmm::device_async_resource_ref | cudf::reset_current_device_resource_ref () |
Reset the current device memory resource reference to the initial resource. More... | |