#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 类型。 更多... | |