cuda - 如何将推力::device_vector 转换到原始指针

Question

我有一个推特device_vector。我想将其转换为原始指针，以便可以将其传递给内核。我该怎么办？

thrust::device_vector<int> dv(10);
//CAST TO RAW
kernel<<<bl,tpb>>>(pass raw)

Answer 1

您可以使用thrust::raw_pointer_cast做到这一点。设备 vector 类具有成员函数data，该函数将thrust::device_ptr返回到 vector 所保持的内存中，可以对其进行强制转换，如下所示:

thrust::device_vector<int> dv(10);

int * dv_ptr = thrust::raw_pointer_cast(dv.data());

kernel<<<bl,tpb>>>(dv_ptr)

(免责声明:用浏览器编写，未经编译，未经测试)。推力中包含一个完整的工作示例: unwrap_pointer.cu