获取CUDA推力::transform运算符(）函数内向量的索引

Get index of vector inside CUDA thrust::transform operator() function

本文关键字：函数 向量 索引 CUDA 推力 transform 运算符 获取      更新时间：2023-10-16

在CUDA Thrust变换中，是否可以在函数内部获得传递到运算符（）函数中的向量的索引？

比如说，我们有

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());

如何在运算符（）中获取向量的索引？基本上，我想要一种简单的方法来制作CUDA中的单位矩阵。

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

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

您可以在推力快速启动指南中阅读更多关于这些概念的内容。

EDIT：针对下面的问题，这里有一个工作示例。虽然这实际上并不使用任何device_vector，但如果我们在GPU上执行此操作（使用device_vector），则唯一会生成任何重要GPU活动的活动将是对thrust::transform的调用，即GPU上只有1个"pass"。

（是的，推力：：序列调用也会生成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;
}
相关文章：