为什么更改块和网格大小会对运行时产生如此大的影响?

时间:2013-06-15 00:51:03

标签: c++ c cuda gpu gpgpu

我正在制作一些cuda tutorial将RGBA图片转换为灰度图像。 但我无法弄清楚为什么更改blockSizegridSize可以提升X33时间。

__global__
void rgba_to_greyscale(const uchar4* const rgbaImage,
                       unsigned char* const greyImage,
                       int numRows, int numCols)
{
    int i = blockIdx.x*numCols + threadIdx.x;
    float channelSum = .299f * rgbaImage[i].x + .587f * rgbaImage[i].y + .114f * rgbaImage[i].z;
    greyImage[i]= channelSum;
}

void your_rgba_to_greyscale(const uchar4 * const h_rgbaImage, uchar4 * const d_rgbaImage,
                            unsigned char* const d_greyImage, size_t numRows, size_t numCols)
{
  const dim3 blockSize(numCols, 1, 1);
  const dim3 gridSize(numRows, 1 , 1);
  rgba_to_greyscale<<<gridSize, blockSize>>>(d_rgbaImage, d_greyImage, numRows, numCols);

  cudaDeviceSynchronize(); checkCudaErrors(cudaGetLastError());
}

当我按上述方式设置时:

const dim3 blockSize(numCols, 1, 1);
const dim3 gridSize(numRows, 1 , 1);

我得到Your code executed in 0.030304 ms

当我设置:

 const dim3 blockSize(1, 1, 1);
 const dim3 gridSize(numRows, numCols , 1);

并更新线程函数以使用新索引:

int i = blockIdx.x*numCols + blockIdx.y;

我得到Your code executed in 0.995456 ms

  1. 我希望它是相反的,因为gpu可以计算所有 第二个网格分裂上的像素分别与它有关 缓存一致性问题?为什么我会得到这些结果?
  2. 这个问题的网格和块大小理论上最好的是什么?是否可以在运行时计算它?
  3. 供参考:

    numRows = 313 numCols =557 
    

    技术属性:

    #uname -a && /usr/bin/nvidia-settings -v
        Linux ip-10-16-23-92 3.2.0-39-virtual #62-Ubuntu SMP Thu Feb 28 00:48:27 UTC 2013 x86_64 x86_64 x86_64 GNU/Linux
    
        nvidia-settings:  version 304.54  (buildmeister@swio-display-x86-rhel47-11)
    

1 个答案:

答案 0 :(得分:6)

建议不要使用网格/块配置。第一个是不可扩展的,因为每个块的线程数对于GPU是有限的,因此它最终会因较大的图像大小而失败。第二个是糟糕的选择,因为每个块只有1个线程,不建议使用,因为GPU占用率非常低。您可以通过CUDA工具包附带的GPU Occupancy Calculator进行验证。推荐的块大小应该是GPU warp大小的倍数(16或32),具体取决于GPU。

在您的情况下,2D网格和块大小的通用和可扩展方法将是这样的:

const dim3 blockSize(16, 16, 1);
const dim3 gridSize((numCols + blockSize.x - 1)/blockSize.x, (numRows + blockSize.y - 1)/blockSize.y , 1);

如果您保持在设备的限制范围内,您可以将块大小从16 x 16更改为您喜欢的任何大小。对于计算能力为1.0到1.3的设备,允许每块最多512个线程。对于计算能力2.0以后的设备,此限制为每块1024个线程。

现在,网格和块是2维的,内核中的索引将被修改如下:

int i = blockIdx.x * blockDim.x + threadIdx.x; //Column
int j = blockIdx.y * blockDim.y + threadIdx.y; //Row

int idx = j * numCols + i;

//Don't forget to perform bound checks
if(i>=numCols || j>=numRows) return;

float channelSum = .299f * rgbaImage[idx].x + .587f * rgbaImage[idx].y + .114f *     rgbaImage[idx].z;
greyImage[idx]= channelSum;