[Mlir-commits] [mlir] [MLIR][NVGPU] Move max threads/blocks size to dialect (NFC) (PR #124454)
Matthias Springer
llvmlistbot at llvm.org
Sun Jan 26 01:04:10 PST 2025
================
@@ -22,8 +22,20 @@
#include "mlir/Dialect/NVGPU/IR/NVGPUEnums.h.inc"
+// Maximum warp size
constexpr int kWarpSize = 32;
+// Maximum number of threads in a block and block in a grid
+// https://docs.nvidia.com/cuda/cuda-c-programming-guide/#features-and-technical-specifications-technical-specifications-per-compute-capability
+constexpr int kMaxTotalBlockdim = 1024;
+constexpr int kMaxBlockdimx = 1024;
----------------
matthias-springer wrote:
nit: in camel case they would be `kMaxBlockDimX` etc.
https://github.com/llvm/llvm-project/pull/124454
More information about the Mlir-commits
mailing list