[Mlir-commits] [mlir] [MLIR][NVVM] Add support for dp4a instructions (PR #139043)
Guray Ozen
llvmlistbot at llvm.org
Thu May 8 21:52:16 PDT 2025
================
@@ -3444,6 +3444,68 @@ def NVVM_Tcgen05StOp : NVVM_Op<"tcgen05.st"> {
let hasVerifier = 1;
}
+//===----------------------------------------------------------------------===//
+// NVVM dp4a Op
+//===----------------------------------------------------------------------===//
+
+def DP4aS8 : I32EnumAttrCase<"S8", 1, "s8">;
+def DP4aU8 : I32EnumAttrCase<"U8", 0, "u8">;
+
+def DP4aType : I32EnumAttr<"DP4aType", "NVVM DP4aType",
+ [DP4aS8, DP4aU8]> {
+ let cppNamespace = "::mlir::NVVM";
+ let genSpecializedAttr = 0;
+}
+
+def DP4aTypeAttr : EnumAttr<NVVM_Dialect, DP4aType, "dp4a_type"> {
+ let assemblyFormat = "`<` $value `>`";
+}
+
+def NVVM_Dp4aOp : NVVM_Op<"dot.accumulate.4way"> {
----------------
grypp wrote:
Can we also change the name of OP?
https://github.com/llvm/llvm-project/pull/139043
More information about the Mlir-commits
mailing list