CUDA Thrust：如何使用掩码进行最大减少操作？

问题描述

我有一个很长的双精度向量 x[]。我还有另一个很长的布尔值 xMask[] 向量。它们具有相同的尺寸。我想使用推力来计算 x[] 的最大值，但仅适用于 xMask[] 为真的那些元素。例如：

x =     [1,2,3,4,5,6,7,8]
xMask = [true,false,true,false]

x[] 与 xMask[] 的 Maximum-Reduce 是 7（不是 8，因为 xMask[] 的值是假的）。

我可以在 Thrust 中轻松做到这一点吗？

解决方法

截至目前，Thrust 中没有名为 reduce_if 的函数，这正是您要搜索的。有多种方法可以使用给定的函数执行此操作，哪种方法最适合您的问题可能取决于掩码中 true 与 false 的比率以及它们的分布方式。>

话虽如此，实现这一目标的规范方法是将 transform_reduce 与 zip_iterator 结合使用：

#include <thrust/device_vector.h>
#include <thrust/functional.h>
#include <thrust/transform_reduce.h>
#include <thrust/zip_iterator.h>

int reduce_if(thrust::device_vector<int> const &data,thrust::device_vector<bool> const &mask) {
  return thrust::transform_reduce(
      thrust::make_zip_iterator(thrust::make_tuple(
          data.cbegin(),mask.cbegin())),thrust::make_zip_iterator(thrust::make_tuple(
          data.cend(),mask.cend())),[](const thrust::tuple<int,bool> &elem){
          return thrust::get<1>(elem) ? thrust::get<0>(elem) : 0;
      },thrust::plus<int>{});
}