diff --git a/mlir/include/mlir/IR/EnumAttr.td b/mlir/include/mlir/IR/EnumAttr.td index 9fec28f03ec28..f7b5967437fb9 100644 --- a/mlir/include/mlir/IR/EnumAttr.td +++ b/mlir/include/mlir/IR/EnumAttr.td @@ -384,9 +384,9 @@ class EnumParameter // The op will appear in the IR as `my_dialect.my_op first`. However, the // generic format of the attribute will be `#my_dialect<"enum first">`. Override // the attribute's assembly format as required. -class EnumAttr traits = []> - : AttrDef { +class EnumAttr traits = []> + : AttrDef { let summary = enumInfo.summary; let description = enumInfo.description; @@ -410,7 +410,7 @@ class EnumAttr:$value); // If a mnemonic was provided, use it to generate a custom assembly format. - let mnemonic = name; + let mnemonic = mnemonicArg; // The default assembly format for enum attributes. Selected to best work with // operation assembly formats.