[Mlir-commits] [mlir] [MLIR][NVGPU] Improve and Cleanup verifier of TMA OPs (PR #70923)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Wed Nov 1 08:31:12 PDT 2023
================
@@ -25,6 +25,8 @@ constexpr int kWarpSize = 32;
/// M size of wgmma.mma_async instruction
constexpr int kWgmmaSizeM = 64;
+/// Maximum tensor dimension that TMA supports
+constexpr int kMaxTMATensorDimension = 5;
----------------
qcolombet wrote:
Where does this limitation come from?
Just curious, we don't necessarily need to lift it.
https://github.com/llvm/llvm-project/pull/70923
More information about the Mlir-commits
mailing list