19 #include <rmm/detail/error.hpp>
20 #include <rmm/detail/export.hpp>
25 namespace RMM_NAMESPACE {
59 void* do_allocate(std::size_t bytes, [[maybe_unused]]
cuda_stream_view stream)
override
63 if (bytes == 0) {
return nullptr; }
66 RMM_CUDA_TRY_ALLOC(cudaMallocManaged(&ptr, bytes));
80 void do_deallocate(
void* ptr,
81 [[maybe_unused]] std::size_t bytes,
84 RMM_ASSERT_CUDA_SUCCESS(cudaFree(ptr));
97 [[nodiscard]]
bool do_is_equal(device_memory_resource
const& other)
const noexcept
override
99 return dynamic_cast<managed_memory_resource const*
>(&other) !=
nullptr;
Strongly-typed non-owning wrapper for CUDA streams with default constructor.
Definition: cuda_stream_view.hpp:41
Base class for all libcudf device memory allocation.
Definition: device_memory_resource.hpp:94
device_memory_resource derived class that uses cudaMallocManaged/Free for allocation/deallocation.
Definition: managed_memory_resource.hpp:36
managed_memory_resource(managed_memory_resource &&)=default
Default move constructor.
managed_memory_resource & operator=(managed_memory_resource &&)=default
Default move assignment operator.
managed_memory_resource & operator=(managed_memory_resource const &)=default
Default copy assignment operator.
managed_memory_resource(managed_memory_resource const &)=default
Default copy constructor.