有没有办法在CUDA中初始化block_size？

Question

我使用的代码如

#define BLOCK_SIZE 32

确定内核函数的块大小。

有没有办法在程序运行时检测内核函数的合理块大小？

Answer 1

运行内核时，内部变量gridDim包含块大小。 （它对应于硬件中的特殊只读寄存器。）

它是3D，所以你必须编写gridDim.x，grimDim.y或grimDim.z。