使用 thrust::reduce 计算 8 位整数向量的总和而不会溢出

Question

我有一个 uint8_t 类型的设备向量，如果可能的话，我想使用 thrust::reduce 计算它的总和。问题是我溢出了，因为总和将远大于 255。我认为下面的代码会通过将结果存储为 32 位整数来计算总和，但事实并非如此。有什么好的方法可以实现吗？

uint8_t * flags_d;
...
const int32_t N_CMP_BLOCKS = thrust::reduce( 
    thrust::device_pointer_cast( flags_d ), 
    thrust::device_pointer_cast( flags_d ) + N,
    (int32_t) 0,
    thrust::plus<int32_t>() );

Answer 1

我认为唯一可行的解决方案是使用 thrust::transform_reduce 在归约中的累加操作之前将 8 位输入数据显式转换为 32 位数量。所以我希望是这样的：

#include <thrust/transform_reduce.h>
#include <thrust/functional.h>
#include <thrust/execution_policy.h>

template<typename T1, typename T2>
struct char2int
{
  __host__ __device__ T2 operator()(const T1 &x) const
  {
    return static_cast<T2>(x);
  }
};

int main()
{
  unsigned char data[6] = {128, 100, 200, 102, 101, 123};
  int result = thrust::transform_reduce(thrust::host,
                                        data, data + 6,
                                        char2int<unsigned char,int>(),
                                        0,
                                        thrust::plus<int>());

  std::cout << "Result is " << result << std::endl;
 
  return 0;
}

更符合您的想法。

使用 thrust::reduce 计算 8 位整数向量的总和而不会溢出

Using thrust::reduce to compute the sum over a vector of 8 bit integers without overflow

c++

cuda

thrust