2

我正在为图片上的直方图编写 CUDA 内核，但我不知道如何从内核返回一个数组，并且当其他线程读取它时该数组会改变。有什么可能的解决方案吗？

__global__ void Hist(
    TColor *dst, //input image
    int imageW,
    int imageH,
 int*data
){
    const int ix = blockDim.x * blockIdx.x + threadIdx.x;
    const int iy = blockDim.y * blockIdx.y + threadIdx.y;

if(ix < imageW && iy < imageH)
{
  int pixel = get_red(dst[imageW * (iy) + (ix)]);
                  //this assign specific RED value of image to pixel

  data[pixel] ++; // ?? problem statement ...
 }
}

@para d_dst：输入图像 TColor 等于 float4。

@para 数据：直方图大小的数组 [255]

extern "C" void
cuda_Hist(TColor *d_dst, int imageW, int imageH,int* data) 
{
  dim3 threads(BLOCKDIM_X, BLOCKDIM_Y);
  dim3 grid(iDivUp(imageW, BLOCKDIM_X), iDivUp(imageH, BLOCKDIM_Y));
  Hist<<<grid, threads>>>(d_dst, imageW, imageH, data);
}
4

3 回答 3

5

您看过 SDK 示例吗？“直方图”示例在 CUDA SDK 中提供（当前 3.0 版在 NVIDIA开发者网站上，3.1 版测试版可供注册开发者使用）。

带有示例的文档很好地解释了如何处理求和，或者使用 GPU 上的全局内存原子，或者分别收集每个块的结果，然后进行单独的缩减（在主机或 GPU 上）。

于 2010-06-08T10:29:56.840 回答
2

直方图在使用 CUDA（或一般使用 GPGPU）实现时并不是特别有效 - 通常您需要在共享内存中生成大量部分直方图，然后对它们求和。您可能需要考虑将此特定任务保留在 CPU 上。

于 2010-06-05T09:00:29.120 回答
0

您将不得不使用原子函数来阻止其他线程使用相同的内存，或者使用部分直方图。无论哪种方式，除非输入图像非常大，否则效率不高。

于 2010-06-05T19:51:03.650 回答