cuda - 通过 CUDA Thrust 对具有偶数或奇数索引的元素求和

Question

如果我使用

 float sum = thrust::transform_reduce(d_a.begin(), d_a.end(), conditional_operator(), 0.f, thrust::plus<float>());

我得到满足由 conditional_operator() 提供的条件的所有元素的总和，如 Conditional reduction in CUDA .

但是我只能对元素求和 d_a[0] , d_a[2] , d_a[4] , d_a[6] ，……？

我想过更改条件运算符，但它适用于数组中的元素，而无需任何索引引用。

我能为此做什么？

Answer 1

我可以想到两种方法来解决此类问题:

使用推力 zip 运算符将计数迭代器与输入数据结合起来，并修改现有的仿函数以接受 (index, data) 的元组。当索引符合您的条件时，您可以让仿函数返回数据，否则为零。这将适用于扫描和减少算法

使用推力置换迭代器来收集要求和的数据并将其传递给标准的归约算法。推力开发者有一个例子 strided iterator您可以使用它来解决仅处理输入迭代器中的每第 n 个条目的问题。

可能值得同时实现并对其进行基准测试以查看哪种方法更快。