CUDA编程模型系列一(核心函数)

最新推荐文章于 2024-09-11 14:42:31 发布

扫地的小何尚

最新推荐文章于 2024-09-11 14:42:31 发布

阅读量1.3k

点赞数 3

文章标签： NVIDIA 人工智能计算机视觉 CUDA 深度学习

本文链接：https://blog.csdn.net/kunhe0512/article/details/131357383

版权

本文是CUDA编程模型系列的第一部分，讲解了如何使用核心函数`__global__`定义GPU上的计算kernel，并通过示例展示了线程块(block)和网格(grid)的概念，以及如何在主机调用并同步kernel。

摘要由CSDN通过智能技术生成

CUDA编程模型系列一(核心函数)

本系列教程将介绍具体的CUDA编程代码的细节

CUDA编程模型系列一(核心函数)

#include <stdio.h>

__global__ void my_first_kernel()
{
    int tidx = threadIdx.x;
    int tidy = threadIdx.y;
    int bidx = blockIdx.x;
    int bidy = blockIdx.y;

    printf("Hello World from thread(thread index:(%d,%d), block index:(%d,%d))!\n", tidy, tidx, bidy, bidx);
}

// thread  --> block --> grid
// SM stream multi-processor
// total threads: block_size * grid_size
int main()
{
    printf("Hello World from CPU!\n");

    dim3 block_size(3,3);
    //t00, t01, t02
    //t10, t11, t12
    //t20, t21, t22
    dim3 grid_size(2,2);
    //b00, b01
    //b10, b11

    my_first_kernel<<<grid_size,block_size>>>();
    cudaDeviceSynchronize();

    return 0;
}