如何在 OpenCL 中将元素添加到数组的末尾？

Question

如何在 OpenCL 中安全地将元素添加到数组的末尾？

安全，我的意思是不存在并发问题，例如一个线程试图将一个元素添加到另一个元素所在的位置

Answer 1

这段代码将使用原子操作将元素从一个数组并行添加到另一个数组以确保安全。

/*
 * list: is of size some size greater than a, one thread per element of list
 * a: is of size "size", initially 0
 * size: this is the size of array "a", initial value is 0
 * capacity: this is the number of elements allocated for "a"
 */
__kernel void AddElementsToEndOfArray(
        __global int* list, 
        __global int* a, 
        __global int size,
        __global int capacity)
{
    local int sz = atomic_add(&(size),1);
    if (sz >= capacity)
        return;

    unsigned int i = get_global_id(0);
    a[sz] = list[i];
}