用计数器替换blockId

时间:2016-07-05 20:44:11

标签: parallel-processing cuda atomic nvidia

最初我在我的代码中使用了blockIdx.x,但我想删除它,而是有一个全局值,并在我的块中使用它而不是blockidx.x。由于我的代码太大而且当我以大输入大小运行它时它会挂起,我认为可以帮助。我原子地增加计数器,但是当我运行代码时它会挂起。谁能看看我的代码,看看我做错了什么?

__device__ int counter = 0;

__global__ void kernel(int * ginput, int * goutput)
{
  const int tid = threadIdx.x;
  const int id = threadIdx.x + blockIdx.x * blockDim.x;
  in myval = ginput[id];  

  if (tid == 0) {
    atomicAdd(&counter, 1);
  }

  __syncthreads();
  if (counter == 0) {
    goutput[tid] = ...;
  }
  if (counter > 0) {
   ...
  }

}

如果我在我的代码中使用blockIdx.x而不是counter,它可以工作,但我只是想用计数器替换它

1 个答案:

答案 0 :(得分:1)

如果您希望counter替换blockIdx.x的使用(即您希望每个块都具有从counter读取的唯一值),那么此类内容应该有效:

__device__ int counter = 0;

__global__ void kernel(int * ginput, int * goutput)
{
  const int tid = threadIdx.x;
  const int id = threadIdx.x + blockIdx.x * blockDim.x;
  __shared__ int my_block_id;


  if (tid == 0) {
    my_block_id = atomicAdd(&counter, 1);
  }

  __syncthreads();
  if (my_block_id == 0) {
    goutput[tid] = ...;
  }
  if (my_block_id > 0) {
   ...
  }

}

你的方法会很麻烦,因为如果你做这样的事情:

if (counter > 5) ....

您可能正在从全局内存中读取counter的新更新值,并且任何数量的块都可能已更新该值,因此行为将无法预测。