boost::compute 流压缩

boost::compute stream compaction

如何使用 boost::compute 进行流压缩?

例如如果您只想对数组中的某些元素执行繁重的操作。首先生成掩码数组,其中包含与要执行操作的元素相对应的数组:

mask = [0 0 0 1 1 0 1 0 1]

然后对mask数组进行互斥扫描(前缀和)得到:

scan = [0 0 0 0 1 2 2 3 3]

然后压缩这个数组:

if (mask[i])
    inds[scan[i]] = i;

获取压缩索引 (inds) 的最终数组:

[3 4 6 8]

最终数组的大小为scan.last() + mask.last()

#include <boost/compute/algorithm/copy_if.hpp>

using namespace boost::compute;

detail::copy_index_if(mask.begin(), mask.end(), inds.begin(), _1 == 1, queue);