[mlir] Add baseAttr to TypedArrayAttrBase.

Element type is useful in code generators but it is lost after construction.

Differential Revision: https://reviews.llvm.org/D72888

diff --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index c19c0b4be37c36f9dddac984a38c9dfa652c729a..eee059e76f33b4261f6a0c5b58824bf8ed3c5e83 100644 (file)
--- a/mlir/include/mlir/IR/OpBase.td
+++ b/mlir/include/mlir/IR/OpBase.td
@@ -1091,6 +1091,8 @@ class TypedArrayAttrBase<Attr element, string description>: ArrayAttrBase<
                           "; })">]>,
     description> {
   let constBuilderCall = "$_builder.getArrayAttr($0)";
+
+  let baseAttr = element;
 }
 
 def I32ArrayAttr : TypedArrayAttrBase<I32Attr,