-1

我想生成一个重复递增整数的序列，给定一个包含每个所需重复次数的列表：

thrust::device_vector<int> reps {3, 2, 5, 1};
//This vector should yield a resulting list:
// {0,0,0, 1,1, 2,2,2,2,2, 3}

理想情况下，我想使用推力 API 来做到这一点。

我考虑过这种方法：

  1. 对代表列表加前缀以生成累积代表列表。
  2. 使用累积代表列表中的最后一个元素分配结果整数向量。
  3. 使用内核，为 reps 列表的每个元素运行一个线程，然后从 循环i = 0 : reps[tid]，存储tid在cumulative_reps[tid]+i。

这会起作用，但最终可能会以串行方式完成大部分工作，并破坏使用 CUDA 的意义。

我想知道是否有推力迭代器和算法的组合来简洁地生成整数列表？或者，即使没有推力，比我概述的方法更好的方法也会很棒。

4

1 回答 1

2

您可以使用与您类似的方法纯粹通过推力来做到这一点。

  1. 对输入进行前缀求和以确定步骤 2 的结果大小，以及步骤 3 的分散索引
  2. 创建一个输出向量来保存结果
  3. 将一个分散到输出向量中的适当位置，由步骤 1 中的索引给出
  4. 对输出向量进行前缀求和。

请注意，如果允许输入代表向量包含 0 值，则必须修改此方法。

这是一个工作示例：

$ cat t404.cu
#include <thrust/scan.h>
#include <thrust/device_vector.h>
#include <thrust/copy.h>
#include <thrust/iterator/constant_iterator.h>
#include <thrust/iterator/permutation_iterator.h>
#include <iostream>

int main(){

  int host_reps[] = {3, 2, 5, 1};
  int ds = sizeof(host_reps)/sizeof(int);
  thrust::device_vector<int> reps(host_reps, host_reps+ds);
  thrust::inclusive_scan(reps.begin(), reps.end(), reps.begin());
  thrust::device_vector<int> result(reps[reps.size()-1]);
  thrust::copy_n(thrust::constant_iterator<int>(1), reps.size()-1, thrust::make_permutation_iterator(result.begin(), reps.begin()));
  thrust::inclusive_scan(result.begin(), result.end(), result.begin());
  thrust::copy_n(result.begin(), result.size(), std::ostream_iterator<int>(std::cout, ","));
  std::cout << std::endl;
}
$ nvcc -o t404 t404.cu
$ ./t404
0,0,0,1,1,2,2,2,2,2,3,
$
于 2019-03-01T13:13:16.620 回答