boost - Boost::compute向量部分和的最快方法是什么

Question

我有一个 10M 浮点数的向量。我想知道每 100 个元素的总和，所以总共有 10000 个总和。最快的方法是什么？

Answer 1

我建议使用reduce_by_key算法、花哨的迭代器和 Boost.Compute lambda expr。每 100 个元素用相同的键标记并减少。我不确定您是否可以替换keys_output为 adiscard_iterator以节省一些性能。

boost::compute::vector<int> keys_output(values_input.size()/100, context);
boost::compute::vector<int> values_output(values_input.size()/100, context);

boost::compute::reduce_by_key(
    boost::compute::make_transform_iterator(
      boost::compute::make_counting_iterator<int>(0),
      boost::compute::_1 / 100
    ),
    boost::compute::make_transform_iterator(
      boost::compute::make_counting_iterator<int>(values_input.size()),
      boost::compute::_1 / 100
    ),
    values_input.begin(),
    keys_output.begin(), 
    values_output.begin(),
    queue
 );