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

Question

我使用的码等

#define BLOCK_SIZE 32 

以确定核函数块大小。

有什么方法可以在程序运行时检测内核函数的合理块大小吗？

Answer 1

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

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