有沒有辦法在CUDA中初始化block_size？

Question

我使用的碼等

#define BLOCK_SIZE 32 

以確定核函數塊大小。

有什麼方法可以在程序運行時檢測內核函數的合理塊大小嗎？

Answer 1

運行內核時，內部變量gridDim包含塊大小。 （它對應於硬件中的特殊只讀寄存器。）

它是3D的，所以您必須編寫gridDim.x，grimDim.y或grimDim.z。