在CUDA内核中声明变量

Question

假设您在CUDA内核中声明一个新变量，然后在多个线程中使用它，例如：

__global__ void kernel(float* delt, float* deltb) {
int i = blockIdx.x * blockDim.x + threadIdx.x;
float a;
a = delt[i] + deltb[i];
a += 1;
}

并且内核调用类似于下面的内容，包含多个线程和块：

int threads = 200;
uint3 blocks = make_uint3(200,1,1);
kernel<<<blocks,threads>>>(d_delt, d_deltb);

1. “a”存储在堆栈上吗？
2. 初始化时是否为每个线程创建了新的“a”？
3. 或者每个线程是否会在未知时间独立访问“a”，可能会弄乱算法？

Answer 1

在内核函数内声明的任何变量（标量或数组），没有extern说明符，对每个线程都是本地的，即每个线程都有自己的变量“副本”，线程之间不会发生数据竞争！

编译器选择局部变量是驻留在寄存器上还是驻留在本地内存中（实际上是全局内存），具体取决于编译器执行的转换和优化。

有关哪些变量在本地内存上的详细信息，请参阅NVIDIA CUDA用户指南，chapter 5.3.2.2

Answer 2

以上都不是。 CUDA编译器足够智能且具有足够的积极性，可以检测到a未被使用，并且可以优化完整的代码。您可以通过使用-Xptxas=-v作为选项编译内核来确认这一点。查看资源计数，它应该基本上没有寄存器，也没有本地内存或堆。

在一个不太重要的例子中，a可能存储在每个线程寄存器中，或者存储在每线程本地存储器中，这是一个片外DRAM。