cuda - 有没有办法为共享内存数组设置默认值？

Question

考虑以下代码:

__global__ void kernel(int *something) {
extern __shared__ int shared_array[];     

// Some operations on shared_array here.

}

是否可以将整个 shared_array 设置为某个值 - 例如0 - 没有明确寻址某个线程中的每个单元格？

Answer 1

您可以像这样并行有效地初始化共享数组

// if SHARED_SIZE == blockDim.x, eliminate this loop
for (int i = threadIdx.x; i < SHARED_SIZE; i += blockDim.x) 
    shared_array[i] = INITIAL_VALUE;
__syncthreads();