通过引用将推力::device_vector 传递给 function

Question

我正在尝试传递结构的device_vector

struct point 
{
    unsigned int x;
    unsigned int y;
}

以下列方式发送到 function：

void print(thrust::device_vector<point> &points, unsigned int index)
{
    std::cout << points[index].y << points[index].y << std::endl;
}

myvector 已正确初始化

print(myvector, 0);

我收到以下错误：

error: class "thrust::device_reference<point>" has no member "x"
error: class "thrust::device_reference<point>" has no member "y"

它出什么问题了？

Answer 1

不幸的是， device_reference<T>不能公开T的成员，但它可以转换为T 。

要实现print ，通过将每个元素转换为临时temp来制作每个元素的临时副本：

void print(thrust::device_vector<point> &points, unsigned int index)
{
    point temp = points[index];
    std::cout << temp.y << temp.y << std::endl;
}

每次调用print时，都会导致从 GPU 到系统 memory 的传输以创建临时文件。 如果您需要一次打印整个points集合，一种更有效的方法是将整个向量points整体复制到host_vector或std::vector （使用thrust::copy ），然后像往常一样遍历集合。

Answer 2

从http://thrust.googlecode.com/svn/tags/1.1.0/doc/html/structthrust_1_1device__reference.html ：

device_reference 充当对存储在设备 memory 中的 object 的引用。 device_reference 不打算直接使用； 相反，这种类型是延迟 device_ptr 的结果。 类似地，获取 device_reference 的地址会产生一个 device_ptr。

也许你需要类似的东西

(&points[index]).get()->x

代替

points[index].x

这有点难看，但是 CUDA 需要一种机制来在 RAM 和 GPU 之间传输数据。