boost :: compute stream compaction

Question

Как сделать сжатие потока с boost :: compute?

E.g. если вы хотите выполнять тяжелую работу только для определенных элементов массива. Сначала вы создаете массив маски с теми, соответствующими элементами, для которых вы хотите выполнить операцию:

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

Затем выполнить эксклюзивное сканирование (префикс суммы) массив маски, чтобы получить:

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()

Answer 1

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

using namespace boost::compute; 

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