• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1// RUN: mlir-opt -allow-unregistered-dialect %s -pass-pipeline='func(canonicalize)' | FileCheck %s
2
3// CHECK-LABEL: func @remove_op_with_inner_ops_pattern
4func @remove_op_with_inner_ops_pattern() {
5  // CHECK-NEXT: return
6  "test.op_with_region_pattern"() ({
7    "foo.op_with_region_terminator"() : () -> ()
8  }) : () -> ()
9  return
10}
11
12// CHECK-LABEL: func @remove_op_with_inner_ops_fold_no_side_effect
13func @remove_op_with_inner_ops_fold_no_side_effect() {
14  // CHECK-NEXT: return
15  "test.op_with_region_fold_no_side_effect"() ({
16    "foo.op_with_region_terminator"() : () -> ()
17  }) : () -> ()
18  return
19}
20
21// CHECK-LABEL: func @remove_op_with_inner_ops_fold
22// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32)
23func @remove_op_with_inner_ops_fold(%arg0 : i32) -> (i32) {
24  // CHECK-NEXT: return %[[ARG_0]]
25  %0 = "test.op_with_region_fold"(%arg0) ({
26    "foo.op_with_region_terminator"() : () -> ()
27  }) : (i32) -> (i32)
28  return %0 : i32
29}
30
31// CHECK-LABEL: func @remove_op_with_variadic_results_and_folder
32// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32, %[[ARG_1:[a-z0-9]*]]: i32)
33func @remove_op_with_variadic_results_and_folder(%arg0 : i32, %arg1 : i32) -> (i32, i32) {
34  // CHECK-NEXT: return %[[ARG_0]], %[[ARG_1]]
35  %0, %1 = "test.op_with_variadic_results_and_folder"(%arg0, %arg1) : (i32, i32) -> (i32, i32)
36  return %0, %1 : i32, i32
37}
38
39// CHECK-LABEL: func @test_commutative_multi
40// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32, %[[ARG_1:[a-z0-9]*]]: i32)
41func @test_commutative_multi(%arg0: i32, %arg1: i32) -> (i32, i32) {
42  // CHECK: %[[C42:.*]] = constant 42 : i32
43  %c42_i32 = constant 42 : i32
44  // CHECK: %[[C43:.*]] = constant 43 : i32
45  %c43_i32 = constant 43 : i32
46  // CHECK-NEXT: %[[O0:.*]] = "test.op_commutative"(%[[ARG_0]], %[[ARG_1]], %[[C42]], %[[C43]]) : (i32, i32, i32, i32) -> i32
47  %y = "test.op_commutative"(%c42_i32, %arg0, %arg1, %c43_i32) : (i32, i32, i32, i32) -> i32
48
49  // CHECK-NEXT: %[[O1:.*]] = "test.op_commutative"(%[[ARG_0]], %[[ARG_1]], %[[C42]], %[[C43]]) : (i32, i32, i32, i32) -> i32
50  %z = "test.op_commutative"(%arg0, %c42_i32, %c43_i32, %arg1): (i32, i32, i32, i32) -> i32
51  // CHECK-NEXT: return %[[O0]], %[[O1]]
52  return %y, %z: i32, i32
53}
54