获取CUDA thrust :: transform operator()函数内的向量索引

时间:2013-07-05 08:52:31

标签: c++ cuda thrust

在CUDA Thrust变换中,是否可以获取向量的索引,传递给函数内的operator()函数?

说,我们有,

struct op{
    float operator()(const float& f){
        //do something like return the index
    }
};
vector<float> v(100);
thrust::transform(v.begin(),v.end(),v.begin(),op());

如何在operator()中获取向量的索引?基本上我想要一种简单的方法在CUDA中制作一个单位矩阵。

1 个答案:

答案 0 :(得分:6)

可能有很多方法可以做到这一点。一种方法是:

  1. 使用thrust::sequence创建与数据向量长度相同的索引向量(或者只使用counting_iterator
  2. 使用zip_iterator返回thrust::tuple,合并数据向量和索引向量,返回数据项的元组及其索引
  3. 定义运算符op()以将特定元组作为其中一个参数
  4. 在运算符中,使用thrust::get<>从元组中检索数据元素或根据需要索引
  5. 您可以在thrust quickstart guide

    中详细了解这些概念

    编辑:在回答下面的问题时,这是一个有效的例子。虽然这实际上并没有使用任何device_vector,但如果我们在GPU上执行此操作(使用device_vector),那么生成任何重要GPU活动的唯一活动就是调用thrust::transform ,即。 GPU上只有1个“通行证”。

    (是的,thrust :: sequence调用也会生成GPU内核,但我只是用它来为这个例子创建一些数据)。

    #include <thrust/host_vector.h>
    #include <thrust/iterator/zip_iterator.h>
    #include <thrust/iterator/counting_iterator.h>
    #include <thrust/sequence.h>
    #include <thrust/copy.h>
    
    #define N 30
    #define SELECT 3
    
    typedef thrust::tuple<int, int>            tpl2int;
    typedef thrust::host_vector<int>::iterator intiter;
    typedef thrust::counting_iterator<int>     countiter;
    typedef thrust::tuple<intiter, countiter>  tpl2intiter;
    typedef thrust::zip_iterator<tpl2intiter>  idxzip;
    
    
    
    struct select_unary_op : public thrust::unary_function<tpl2int, int>
    {
      __host__ __device__
      int operator()(const tpl2int& x) const
      {
        if ((x.get<1>() %SELECT) == 0)
          return x.get<0>();
        else return -1;
       }
    };
    
    int main() {
    
      thrust::host_vector<int> A(N);
      thrust::host_vector<int> result(N);
      thrust::sequence(A.begin(), A.end());
      thrust::counting_iterator<int> idxfirst(0);
      thrust::counting_iterator<int> idxlast = idxfirst +N;
    
      idxzip first = thrust::make_zip_iterator(thrust::make_tuple(A.begin(), idxfirst));
      idxzip  last = thrust::make_zip_iterator(thrust::make_tuple(A.end(), idxlast));
      select_unary_op my_unary_op;
    
      thrust::transform(first, last, result.begin(), my_unary_op);
      std::cout << "Results :" << std::endl;
      thrust::copy(result.begin(), result.end(), std::ostream_iterator<int>( std::cout, " "));
      std::cout << std::endl;
    
    
      return 0;
    
    }