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);
如何使用 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);