这篇博客详细解析了CUDA编程中线程块在多维grid中的坐标表示，以及如何将高维坐标转换为一维ID。内容涵盖1Dgrid,1Dblock, 3Dgrid,1Dblock, 1Dgrid,2Dblock和3Dgrid,3Dblock等场景下的坐标转换公式，帮助理解GPU并行计算的底层原理。
首先引入一个栗子：被除数 = 除数 * 商 + 余数
类别以下, 用公式表示：最终的线程Id = blockId * blockSize + threadId
- blockId ：当前 block 在 grid 中的坐标（可能是1维到3维）
- blockSize ：block 的大小，描述其中含有多少个 thread
- threadId ：当前 thread 在 block 中的坐标（同样从1维到3维）
下面先理清几个关键点：
grid 中 含有若干个 blocks，其中 blocks 的数量由 gridDim.x/y/z 来描述。某个 block 在此 grid 中的坐标由 blockIdx.x/y/z 描述。
blocks 中含有若干个 threads，其中 threads 的数量由 blockDim.x/y/z 来描述。某个 thread 在此 block 中的坐标由 threadIdx.x/y/z 描述。
接着一个多维的坐标如何用一维数据表达呢？这里大家想一想两位数和三位数，就是很好的例子。数字 = 百位数字 * 100 + 十位数字 * 10 + 个位数字。
当我们得知每个维度上的大小时，就可以利用这样的进制将三维坐标转换为1维坐标。
一般来说坐标（x, y, z）分别所在的维度大小是（Dx, Dy, Dz），一般会把 z 看成高纬度，接着是 y ，最后是 x。
高维度坐标转一维坐标公式 id = Dx * Dy * z + Dx *
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
