#include <rmm/detail/export.hpp>#include <rmm/mr/device/device_memory_resource.hpp>#include <cstddef>#include <functional>#include <utility>callback_memory_resource.hpp 的包含依赖图
类 | |
| 类 | rmm::mr::callback_memory_resource |
| 一种设备内存资源，使用提供的回调函数进行内存分配和释放。 更多... | |
类型定义 | |
| 使用 | rmm::mr::allocate_callback_t = std::function< void *(std::size_t, cuda_stream_view, void *)> |
| 回调函数类型，由回调内存资源用于分配。更多... | |
| 使用 | rmm::mr::deallocate_callback_t = std::function< void(void *, std::size_t, cuda_stream_view, void *)> |
| 回调函数类型，由 callback_memory_resource 用于释放。更多... | |
