Add TabelGen support to logically AND a list of attribute constraints.

    This is the AllOf version for AttrConstraint and allows specifying the combined constraint during pattern match.

--

PiperOrigin-RevId: 242540243

diff --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index baaaa27..52a4a1e 100644 (file)
--- a/mlir/include/mlir/IR/OpBase.td
+++ b/mlir/include/mlir/IR/OpBase.td
@@ -637,6 +637,15 @@ class Confined<Attr attr, list<AttrConstraint> constraints> : Attr<
   let isOptional = attr.isOptional;
 }
 
+// An AttrConstraint that holds if all attr constraints specified in
+// 'constraints' hold.
+class AllAttrConstraintsOf<list<AttrConstraint> constraints> : AttrConstraint<
+    AllOf<!listconcat([!head(constraints).predicate],
+                      !foreach(pred, !tail(constraints), pred.predicate))>,
+    !foldl(/*init*/!head(constraints).description, /*list*/!tail(constraints),
+           prev, cur, prev # " and " # cur.description)> {
+}
+
 class IntMinValue<int n> : AttrConstraint<
     CPred<"{0}.cast<IntegerAttr>().getInt() >= " # n>,
     "whose minimal value is " # n>;

diff --git a/mlir/test/mlir-tblgen/pattern-allof-attr.td b/mlir/test/mlir-tblgen/pattern-allof-attr.td
new file mode 100644 (file)
index 0000000..e6f3799
--- /dev/null
+++ b/mlir/test/mlir-tblgen/pattern-allof-attr.td
@@ -0,0 +1,35 @@
+// RUN: mlir-tblgen -gen-rewriters -I %S/../../include %s | FileCheck %s
+
+include "mlir/IR/OpBase.td"
+
+def FirstConstraint : AttrConstraint<CPred<"FirstConstraint">,
+  "first constraint">;
+def SecondConstraint : AttrConstraint<CPred<"SecondConstraint">,
+  "second constraint">;
+def ThirdConstraint : AttrConstraint<CPred<"ThirdConstraint">,
+  "third constraint">;
+
+def OpA : Op<"op_a", []> {
+  let arguments = (ins
+    I32Attr:$attr
+  );
+
+  let results = (outs I32:$result);
+}
+
+def : Pat<
+  (OpA AllAttrConstraintsOf<
+    [FirstConstraint,
+     SecondConstraint,
+     ThirdConstraint]>:$more),
+  (OpA $more)>;
+
+// Test combining AttrConstraint during pattern match.
+// ---
+
+// CHECK-LABEL: struct GeneratedConvert0
+
+// CHECK:      auto attr = op0->getAttrOfType<IntegerAttr>("attr");
+// CHECK: if (!(((FirstConstraint)) && ((SecondConstraint)) && ((ThirdConstraint)))) return matchFailure();
+// CHECK-NEXT: s.more = attr;
+