cuda - CUDA 三方括号可选参数

Question

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

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

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

Answer 1

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

extern __shared__ T smem[];

代替

__shared__ T[CONSTANT];

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