[Mlir-commits] [mlir] [MLIR][NVVM] Add tcgen05.mma MLIR Ops (PR #164356)

Durgadoss R llvmlistbot at llvm.org
Mon Oct 20 23:34:48 PDT 2025


================
@@ -4537,6 +4537,645 @@ def NVVM_ClusterLaunchControlQueryCancelOp
   }];
 }
 
+//===----------------------------------------------------------------------===//
+// NVVM tcgen05.mma attributes
+//===----------------------------------------------------------------------===//
+
+def Tcgen05MMAKindF16          : I32EnumAttrCase<"F16",    0, "f16">;
+def Tcgen05MMAKindTF32         : I32EnumAttrCase<"TF32",   1, "tf32">;
+def Tcgen05MMAKindF8F6F4       : I32EnumAttrCase<"F8F6F4", 2, "f8f6f4">;
+def Tcgen05MMAKindINT8         : I32EnumAttrCase<"I8",     3, "i8">;
----------------
durga4github wrote:

ah, sometimes it is a mix of types like "f8f6f4", which does not have any equivalent individual type representation. So, I suppose we need to use a separate `Attr` for this case.

https://github.com/llvm/llvm-project/pull/164356


More information about the Mlir-commits mailing list