#include <rmm/detail/export.hpp>
#include <rmm/detail/thrust_namespace.h>
#include <rmm/mr/device/thrust_allocator_adaptor.hpp>
#include <thrust/device_vector.h>
类型定义 | |
template<typename T > | |
using | rmm::device_vector = thrust::device_vector< T, rmm::mr::thrust_allocator< T > > |
使用 RMM 进行内存分配的 thrust::device_vector 的别名。 | |