我正在制作一些cuda tutorial将RGBA图片转换为灰度图像。
但我无法弄清楚为什么更改blockSize
和gridSize
可以提升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
。
供参考:
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)
答案 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;