为何thrust::count与thrust::count_if统计设备向量零元素结果不同？

问题描述 #

我有一个包含16,679,056个float元素的Thrust设备向量：

• 使用thrust::count统计零元素：int result_count_0 = thrust::count(d_result.begin(),d_result.end(), 0);，结果为14,786,283。
• 使用自定义结构体的thrust::count_if统计：

struct is_zero
{
  __host__ __device__
  bool operator()(int x)
  {
    return x == 0;
  }
};

// 调用代码
is_zero iszero;
int result_count_isZero = thrust::count_if(d_result.begin(),d_result.end(),iszero);

结果为15,733,822，且该结果与非零元素数量（945,234）相加刚好等于向量总长度，逻辑更合理。

核心原因：类型不匹配导致的比较逻辑差异 #

两种统计方式的本质差异在于元素类型转换规则和比较逻辑的不同：

1. thrust::count的比较逻辑
   传入的比较值是整数0，会被隐式转换为float类型的0.0f，此时会严格进行浮点数相等比较：只有当向量中的float元素精确等于0.0f（或负零-0.0f）时，才会被统计为零。而那些由于计算精度误差产生的极小值（如1e-30f、0.999f等），因为和0.0f不相等，不会被计入统计结果。

2. thrust::count_if的比较逻辑
   自定义结构体的operator()参数是int类型，Thrust会自动将向量中的float元素隐式转换为int：根据C++类型转换规则，所有绝对值小于1的float值（包括正的极小值、负的极小值，甚至0.999f这类接近1的数）都会被截断为int类型的0，因此这些值都会被判定为零并统计，最终结果包含了所有“转int后为0”的元素，所以数量更多，且与非零元素总数匹配。

验证与修正建议 #

如果想让两种统计方式结果一致，需要统一比较逻辑：

• 修改count_if的结构体，使用float参数明确转换逻辑：

struct is_zero
{
  __host__ __device__
  bool operator()(float x)
  {
    return static_cast<int>(x) == 0;
  }
};

• 若要让thrust::count实现相同逻辑，需借助count_if完成，因为thrust::count仅支持直接的相等比较，无法自定义类型转换规则。

内容的提问来源于stack exchange，提问作者Michailina