c++ - 如何通过推力将主机 vector 复制到设备 vector

Question

我想将主机 std::vector 复制到 thrust::device_vector

std::vector<double> p_a(100)
thrust::device_vector<double> d_vec

我想将p_a复制到d_vec

Answer 1

来自documention ,

您可以使用此构造函数:

__host__ thrust::device_vector< T, Alloc >::device_vector   
(   const std::vector< OtherT, OtherAlloc > &   v   )   

Copy constructor copies from an exemplar std::vector.

此构造函数接收要复制的 std::vector 作为参数。 所以，你可以这样做:

std::vector<double> p_a(100);
thrust::device_vector<double> d_vec(p_a);

您也可以使用复制分配:

d_vec = p_a;