GPU CUDA编程中threadIdx, blockIdx, blockDim, gridDim之间的区别与联系

  前期写代码的时候都会困惑这个实际的threadIdx(tid,实际的线程id)到底是多少,自己写出来的对不对,今天经过自己一些小例子的推敲,以及找到官网的相关介绍,总算自己弄清楚了。

      在启动kernel的时候,要通过指定gridsize和blocksize才行,举下面的例子说说:

     dim3 gridsize(2,2);

     dim3 blocksize(4,4);

     gridsize相当于是一个2*2的block,gridDim.x,gridDim.y,gridDim.z相当于这个dim3的x,y,z方向的维度,这里是2*2*1。序号从0到3,且是从上到下的顺序,就是说是下面的情况:

     grid中的blockidx序号标注情况为:            0     2 

                                                                        1     3

    blocksize则是指里面的thread的情况,blockDim.x,blockDim.y,blockDim.z相当于这个dim3的x,y,z方向的维度,这里是4*4*1.序号是0-15,也是从上到下的标注:

    block中的threadidx序号标注情况:          0      4       8      12 

                                                                      1       5       9       13

                                                                      2       6       10     14

                                                                      3       7       11      15

  应该这样子就一目了然了,然后求实际的tid的时候:

最后还发现了一个2D * 2D 表示threadid的二维位置的标示图, 适用于将每个threadid跟矩阵中的二维元素进行一一对应。

 

引文链接:threadIdx,blockIdx, blockDim, gridDim之间的区别与联系

posted @   rainbow70626  阅读(19509)  评论(0编辑  收藏  举报
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示