xref: /llvm-project/mlir/test/mlir-pdll/Parser/rewrite.pdll (revision faf42264e5401a1dfca95b701e5c2bf951d7f8a7)
1// RUN: mlir-pdll %s -I %S -split-input-file | FileCheck %s
2
3// CHECK:  Module
4// CHECK:  `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Tuple<>>
5Rewrite Foo();
6
7// -----
8
9// CHECK:  Module
10// CHECK:  `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Tuple<>> Code< /* Native Code */ >
11Rewrite Foo() [{ /* Native Code */ }];
12
13// -----
14
15// CHECK: Module
16// CHECK: `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Value>
17// CHECK:   `Inputs`
18// CHECK:     `-VariableDecl {{.*}} Name<arg> Type<Op>
19// CHECK:   `Results`
20// CHECK:     `-VariableDecl {{.*}} Name<> Type<Value>
21// CHECK:   `-CompoundStmt {{.*}}
22// CHECK:     `-ReturnStmt {{.*}}
23// CHECK:       `-MemberAccessExpr {{.*}} Member<$results> Type<Value>
24// CHECK:         `-DeclRefExpr {{.*}} Type<Op>
25// CHECK:           `-VariableDecl {{.*}} Name<arg> Type<Op>
26Rewrite Foo(arg: Op) -> Value => arg;
27
28// -----
29
30// CHECK: Module
31// CHECK: `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Tuple<result1: Value, result2: Attr>>
32// CHECK:   `Results`
33// CHECK:     |-VariableDecl {{.*}} Name<result1> Type<Value>
34// CHECK:     | `Constraints`
35// CHECK:     |   `-ValueConstraintDecl {{.*}}
36// CHECK:     `-VariableDecl {{.*}} Name<result2> Type<Attr>
37// CHECK:       `Constraints`
38// CHECK:         `-AttrConstraintDecl {{.*}}
39// CHECK:   `-CompoundStmt {{.*}}
40// CHECK:     `-ReturnStmt {{.*}}
41// CHECK:       `-TupleExpr {{.*}} Type<Tuple<result1: Value, result2: Attr>>
42// CHECK:         |-MemberAccessExpr {{.*}} Member<0> Type<Value>
43// CHECK:         | `-TupleExpr {{.*}} Type<Tuple<Value, Attr>>
44// CHECK:         `-MemberAccessExpr {{.*}} Member<1> Type<Attr>
45// CHECK:           `-TupleExpr {{.*}} Type<Tuple<Value, Attr>>
46Rewrite Foo() -> (result1: Value, result2: Attr) => (_: Value, attr<"10">);
47
48// -----
49
50// Test that anonymous Rewrites are uniquely named.
51
52// CHECK: Module
53// CHECK: UserRewriteDecl {{.*}} Name<<anonymous_rewrite_0>> ResultType<Tuple<>>
54// CHECK: UserRewriteDecl {{.*}} Name<<anonymous_rewrite_1>> ResultType<Attr>
55Rewrite Outer() {
56  Rewrite() {};
57  Rewrite() => attr<"10">;
58}
59