#include <rmm/cuda_device.hpp>#include <rmm/detail/cuda_memory_resource.hpp>#include <rmm/detail/export.hpp>#include <rmm/detail/thrust_namespace.h>#include <rmm/mr/device/per_device_resource.hpp>#include <rmm/resource_ref.hpp>#include <thrust/device_malloc_allocator.h>#include <thrust/device_ptr.h>#include <thrust/memory.h>

类 | |
| class | rmm::mr::thrust_allocator< T > |
一个与 Thrust 容器和算法兼容的 allocator,它使用 device_async_resource_ref 进行内存分配和释放。 更多... | |
| struct | rmm::mr::thrust_allocator< T >::rebind< U > |
提供使用另一种类型实例化的 thrust_allocator 类型。 更多... | |