如何使用共享内存计算直方图

Question

我想用共享内存计算直方图，我编写如下代码，但我发现cuda和cpu的结果不同，cuda的结果是cpu的blockDim.x次，为什么？ 我该如何解决？ 谢谢！

__global__ void CalHistKernel(int*imgData,int*bins,int datasize)
{
__shared__ int _bins[3];

int tx=threadIdx.x;
int idx=blockIdx.x*blockDim.x+threadIdx.x;//blockDim.y=1
if(tx<3)
{
    _bins[tx]=0;       
}
__syncthreads();

if(idx<datasize)
{       
    atomicAdd((int*)&_bins[imgData[idx]],1);     

}
__syncthreads();
for(int i=0;i<3;i++)   
    atomicAdd((int*)&bins[i],_bins[i]);
}  

Answer 1

看起来它是由于最终循环所致，您将每个共享内存区中的值添加到全局内存区。请记住，循环在每个线程中执行。