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

Question

我想用共享内存计算直方图，我编写了如下代码，但是我发现cuda和cpu的结果不同，cuda的结果是cpu的块Dim.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

看起来这是由于最后一个循环造成的，在这个循环中，您将每个共享内存桶中的值添加到全局内存桶中。记住，循环是在每个线程中执行的。