c++ - 推力 vector 指针声明

Question

有没有办法在不实际分配 vector 的情况下声明推力 vector 指针？我需要将此指针用作类中的成员变量。因为我事先并不知道 vector 的大小，所以我不能将 vector 静态分配为成员变量。

Answer 1

您可以使用:

#include <thrust/device_ptr.h>
#include <thrust/device_vector.h>

template <typename T>
class my_thrust_class
{
  public:
    thrust::device_ptr<T> my_dptr;
}

声明一个设备指针，然后可以将其初始化为您希望它引用的任何 device_vector 的开头:

thrust::device_vector<float> my_vec(3);
my_thrust_class<float> A;
A.my_dptr = my_vec.data();