mlir/test/mlir-pdll/Parser/rewrite.pdll

   1 // RUN: mlir-pdll %s -I %S -split-input-file | FileCheck %s
   2
   3 // CHECK:  Module
   4 // CHECK:  `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Tuple<>>
   5 Rewrite Foo();
   6
   7 // -----
   8
   9 // CHECK:  Module
  10 // CHECK:  `-UserRewriteDecl {{.*}} Name<Foo> ResultType<Tuple<>> Code< /* Native Code */ >
  11 Rewrite 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>
  26 Rewrite 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>>
  46 Rewrite 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>
  55 Rewrite Outer() {
  56   Rewrite() {};
  57   Rewrite() => attr<"10">;
  58 }