如何将 atomicAdd 应用于 CUDA 中数组的每个元素？

Question

我有一个来自CUDA 示例的代码到 atomic 添加单个变量

__global__ void myadd(int *data)
{
  unsigned int x = blockIdx.x;
  unsigned int y = threadIdx.x;
  if ( (x%2==0) && (y%2==1) ) atomicAdd(data,1);
}

int main(void)
{
  int *Hdt;
  Hdt = (int*)malloc(sizeof(int));
  // ... CUDA initialization here
  myadd<<<20, 10>>>(Hdt);
  cudaFree(Hdt);
}

它对我很有用。但是我正在扩展我的代码，所以我想将一个数组而不是一个数字传递给内核

__global__ void myadd(int *data)
{
  unsigned int x = blockIdx.x;
  unsigned int y = threadIdx.x;
  unsigned int z = threadIdx.y;
  if ( (x%2==0) && (y%2==1) && (z>4) ) atomicAdd(data[z],1);
}

int main(void)
{
  int *Hdt;
  Hdt = (int*)malloc(sizeof(20*int));
  // ... CUDA initialization here
  myadd<<<20, dim3(10, 20)>>>(Hdt);
  cudaFree(Hdt);
}

但是它没有编译，错误信息是：

错误：没有重载函数“atomicAdd”的实例与 参数列表参数类型为：(int, int)

Answer 1

替换：

atomicAdd(data[z],1);

与

atomicAdd(&data[z],1);

如果您仔细观察，在第一种情况下，您将指针作为 atomicAdd() 的第一个参数。