18 #include <rmm/detail/error.hpp>
19 #include <rmm/detail/export.hpp>
28 namespace RMM_NAMESPACE {
94 template <
typename Upstream,
typename ExceptionType = rmm::out_of_memory>
110 : upstream_{upstream}, callback_{std::move(callback)}, callback_arg_{callback_arg}
128 callback_{std::move(callback)},
129 callback_arg_{callback_arg}
170 }
catch (exception_type
const& e) {
171 if (!callback_(bytes, callback_arg_)) {
throw; }
184 void do_deallocate(
void* ptr, std::size_t bytes, cuda_stream_view stream)
override
186 get_upstream_resource().deallocate_async(ptr, bytes, stream);
196 [[nodiscard]]
bool do_is_equal(device_memory_resource
const& other)
const noexcept
override
198 if (
this == &other) {
return true; }
199 auto cast =
dynamic_cast<failure_callback_resource_adaptor<Upstream> const*
>(&other);
200 if (cast ==
nullptr) {
return false; }
201 return get_upstream_resource() == cast->get_upstream_resource();
Strongly-typed non-owning wrapper for CUDA streams with default constructor.
Definition: cuda_stream_view.hpp:39
Base class for all librmm device memory allocation.
Definition: device_memory_resource.hpp:93
void * allocate_async(std::size_t bytes, std::size_t alignment, cuda_stream_view stream)
Allocates memory of size at least bytes.
Definition: device_memory_resource.hpp:216
A device memory resource that calls a callback function when allocations throw a specified exception ...
Definition: failure_callback_resource_adaptor.hpp:95
failure_callback_resource_adaptor(device_async_resource_ref upstream, failure_callback_t callback, void *callback_arg)
Construct a new failure_callback_resource_adaptor using upstream to satisfy allocation requests.
Definition: failure_callback_resource_adaptor.hpp:107
ExceptionType exception_type
The type of exception this object catches/throws.
Definition: failure_callback_resource_adaptor.hpp:97
failure_callback_resource_adaptor(Upstream *upstream, failure_callback_t callback, void *callback_arg)
Construct a new failure_callback_resource_adaptor using upstream to satisfy allocation requests.
Definition: failure_callback_resource_adaptor.hpp:124
failure_callback_resource_adaptor(failure_callback_resource_adaptor &&) noexcept=default
Default move constructor.
std::function< bool(std::size_t, void *)> failure_callback_t
Callback function type used by failure_callback_resource_adaptor.
Definition: failure_callback_resource_adaptor.hpp:51
cuda::mr::async_resource_ref< cuda::mr::device_accessible > device_async_resource_ref
Alias for a cuda::mr::async_resource_ref with the property cuda::mr::device_accessible.
Definition: resource_ref.hpp:41
device_async_resource_ref to_device_async_resource_ref_checked(Resource *res)
Convert pointer to memory resource into device_async_resource_ref, checking for nullptr
Definition: resource_ref.hpp:79
Management of per-device device_memory_resources.