#include <rmm/detail/export.hpp>#include <rmm/mr/device/device_memory_resource.hpp>#include <cstddef>#include <functional>#include <utility>
Include dependency graph for callback_memory_resource.hpp:
Go to the source code of this file.
Classes | |
| class | rmm::mr::callback_memory_resource |
| A device memory resource that uses the provided callbacks for memory allocation and deallocation. More... | |
Typedefs | |
| using | rmm::mr::allocate_callback_t = std::function< void *(std::size_t, cuda_stream_view, void *)> |
| Callback function type used by callback memory resource for allocation. More... | |
| using | rmm::mr::deallocate_callback_t = std::function< void(void *, std::size_t, cuda_stream_view, void *)> |
| Callback function type used by callback_memory_resource for deallocation. More... | |
