推力结构矢量的迭代器

时间:2011-07-05 16:04:30

标签: c++ stl iterator cuda thrust

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

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]"

我做错了什么?

2 个答案:

答案 0 :(得分:5)

好的,这个比我想象的要复杂一点:) 以下是我的调查结果:

你的问题来自推力的实施。 Thrust使用名为device_reference的类型,正如其文档所述:http://wiki.thrust.googlecode.com/hg/html/classthrust_1_1device__reference.html

  

device_reference充当   类似于引用的对象到对象   存储在设备存储器中。   device_reference无意   直接使用;相反,这种类型是   引用a的结果   device_ptr。同样,服用了   device_reference的地址产生a   device_ptr

但是,在某些情况下,我们隐含地处理device_reference。例如,当device_reference作为参数传递给等待POD的函数时(或多或少会尝试使用operator<<),会出现以下问题:

  

另一种常见的情况   device_reference不能直接   用来代替它的指示对象   将它们作为参数传递时发生   像printf这样的函数   varargs参数。因为varargs   参数必须是Plain Old Data,a    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;
}

现在,您可以选择自己喜欢的那个!

答案 1 :(得分:4)

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

:))