推力中结构向量的迭代器

iterator for vector of structures in thrust

本文关键字：迭代器 向量 结构      更新时间：2023-10-16

我正试图以这种方式访问向量元素

struct point
{
    unsigned int x;
    unsigned int y;
};
...
thrust::device_vector<point> devPoints(hPoints.begin(), hPoints.end());
for(thrust::device_vector<point>::iterator iter = devPoints.begin(); iter != devPoints.end(); iter++) 
{
    std::cout << iter->x << " " << iter->y << " " << std::endl; (1)
}

device_vector已正确初始化。我得到以下错误：

error: expression must have pointer type (at 1)
error: no suitable user-defined conversion from "const thrust::detail::normal_iterator<thrust::device_ptr<point>>" to "thrust::device_ptr<point>" exists
          detected during instantiation of "Pointer thrust::experimental::iterator_facade<Derived, Pointer, Value, Space, Traversal, Reference, Difference>::operator->() const [with Derived=thrust::detail::normal_iterator<thrust::device_ptr<point>>, Pointer=thrust::device_ptr<point>, Value=point, Space=thrust::detail::cuda_device_space_tag, Traversal=thrust::random_access_traversal_tag, Reference=thrust::device_reference<point>, Difference=ptrdiff_t]"

我做错了什么？

好吧，这个比我预期的要复杂一些：(
以下是我的调查结果：

你的问题来自于推力的实现。Thrust使用一种称为device_reference的类型，正如其文档所述：http://wiki.thrust.googlecode.com/hg/html/classthrust_1_1device__reference.html

device_reference充当对象的类引用对象存储在设备存储器中。device_reference并非直接使用；相反，这种类型是推迟的结果device_ptr。同样device_reference的地址产生device_ptr。

然而，在某些情况下，我们隐式地处理device_reference。例如，当将device_reference作为参数传递给等待POD的函数时(或多或少就是您尝试使用operator<<的方法(，会出现以下问题：

另一种常见情况是device_reference不能直接用于代替其引用对象将它们作为参数传递时发生类似printf的功能varargs参数。因为varargs参数必须是纯旧数据device_reference到POD类型需要铸造当传递到printf:时

话虽如此，你所要做的就是将你的device_reference投射到你正在处理的POD上。在你的情况下，你会这样做：

for(thrust::device_vector<point>::iterator iter = devPoints.begin(); iter != devPoints.end(); iter++)  {
   std::cout << (static_cast<point>(*iter)).x << " " << (static_cast<point>(*iter)).y << std::endl;
}

在我看来，这不是最优雅的解决方案，我宁愿使用std::copy算法来打印point类的内容。因此，我编写了一个小的示例文件，使用point类并使用三种不同的方式打印：

#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <cstdlib>
#include <algorithm>
#include <iostream>
struct point
{
        unsigned int x;
        unsigned int y;
};
__host__
point getRandomPoint() {
        point p;
        p.x = rand();
        p.y = rand();
        return p;
}
__host__
std::ostream& operator<< (std::ostream& os, const point& p) {
        os << "[ " << p.x << " ; " << p.y << " ]";
        return os;
}
int main() {
        // fill the host_vector with random points
        thrust::host_vector<point> hPoints(512);
        thrust::generate(hPoints.begin(), hPoints.end(), getRandomPoint);
        // copy hPoints content to device memory
        thrust::device_vector<point> devPoints(hPoints.begin(), hPoints.end());
        // first way 
        for(thrust::device_vector<point>::iterator iter = devPoints.begin(); iter != devPoints.end(); iter++)  {
                std::cout << (static_cast<point>(*iter)).x << " " << (static_cast<point>(*iter)).y << std::endl;
        }
        // second way
        for(thrust::device_vector<point>::iterator iter = devPoints.begin(); iter != devPoints.end(); iter++)
        {
                std::cout << *iter << std::endl;
        }
        // third way
        std::copy(devPoints.begin(), devPoints.end(), std::ostream_iterator< point >(std::cout, "  $  ") );
        return 0;
}

现在，由你来选择你喜欢的！

std::cout << iter->x << " " << iter->y << " " << std::endl; 
                                      ^^^^

：((

相关文章：