timerring

CUDA Thread Indexing

August 9, 2023 · 18 min read
Tutorial
CUDA | GPU

结合向量相加和图像处理,介绍 CPU 与 GPU 内存操作、线程索引以及 Grid 和 Block 的配置。

我们要处理的问题

如图,有离散型的,例如向量相加等等。有 local 卷积类型的,用其他一片的元素值求得一个元素。还有 All to All 类型的,类似于傅里叶变换。

回顾一下 CUDA 程序的编写:

  1. 把输入数据从 CPU 内存复制到 GPU 显存
  2. 在执行芯片上缓存数据,加载 GPU 程序并执行
  3. 将计算结果从 GPU 显存中复制到 CPU 内存中
...