CUDA 三重括号可选参数

Question

我知道可以使用三方括号语法启动内核:

kernel<<<NumOfBlocks, NumOfThreadsPerBlock, A, B>>> (...);

最后两个参数是可选的，但我想知道它们指的是什么。 B 应该指的是内核应该执行的流，但是 A 呢？

Answer 1

这是共享内存量。您可以在共享内存中传递所需的字节数。在内核中你应该使用

extern __shared__ T smem[];

而不是

__shared__ T[CONSTANT];

在内核中获取大小为 A/sizeof(T) 的共享数组。