ebad5fb309
The canonicalize command-line options currently have no effect, as the pass is reading the pass options in its constructor, before they're actually initialized. This results in the default values of the options always being used. The change here moves the initialization of the `GreedyRewriteConfig` out of the constructor, so that it runs after the pass options have been parsed. Fixes #55466 Reviewed By: rriddle Differential Revision: https://reviews.llvm.org/D125621
105 lines
3.9 KiB
MLIR
105 lines
3.9 KiB
MLIR
// RUN: mlir-opt %s -pass-pipeline='func.func(canonicalize)' | FileCheck %s
|
|
// RUN: mlir-opt %s -pass-pipeline='func.func(canonicalize{region-simplify=false})' | FileCheck %s --check-prefixes=CHECK,NO-RS
|
|
|
|
// CHECK-LABEL: func @remove_op_with_inner_ops_pattern
|
|
func.func @remove_op_with_inner_ops_pattern() {
|
|
// CHECK-NEXT: return
|
|
"test.op_with_region_pattern"() ({
|
|
"test.op_with_region_terminator"() : () -> ()
|
|
}) : () -> ()
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func @remove_op_with_inner_ops_fold_no_side_effect
|
|
func.func @remove_op_with_inner_ops_fold_no_side_effect() {
|
|
// CHECK-NEXT: return
|
|
"test.op_with_region_fold_no_side_effect"() ({
|
|
"test.op_with_region_terminator"() : () -> ()
|
|
}) : () -> ()
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func @remove_op_with_inner_ops_fold
|
|
// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32)
|
|
func.func @remove_op_with_inner_ops_fold(%arg0 : i32) -> (i32) {
|
|
// CHECK-NEXT: return %[[ARG_0]]
|
|
%0 = "test.op_with_region_fold"(%arg0) ({
|
|
"test.op_with_region_terminator"() : () -> ()
|
|
}) : (i32) -> (i32)
|
|
return %0 : i32
|
|
}
|
|
|
|
// CHECK-LABEL: func @remove_op_with_variadic_results_and_folder
|
|
// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32, %[[ARG_1:[a-z0-9]*]]: i32)
|
|
func.func @remove_op_with_variadic_results_and_folder(%arg0 : i32, %arg1 : i32) -> (i32, i32) {
|
|
// CHECK-NEXT: return %[[ARG_0]], %[[ARG_1]]
|
|
%0, %1 = "test.op_with_variadic_results_and_folder"(%arg0, %arg1) : (i32, i32) -> (i32, i32)
|
|
return %0, %1 : i32, i32
|
|
}
|
|
|
|
// CHECK-LABEL: func @test_commutative_multi
|
|
// CHECK-SAME: (%[[ARG_0:[a-z0-9]*]]: i32, %[[ARG_1:[a-z0-9]*]]: i32)
|
|
func.func @test_commutative_multi(%arg0: i32, %arg1: i32) -> (i32, i32) {
|
|
// CHECK-DAG: %[[C42:.*]] = arith.constant 42 : i32
|
|
%c42_i32 = arith.constant 42 : i32
|
|
// CHECK-DAG: %[[C43:.*]] = arith.constant 43 : i32
|
|
%c43_i32 = arith.constant 43 : i32
|
|
// CHECK-NEXT: %[[O0:.*]] = "test.op_commutative"(%[[ARG_0]], %[[ARG_1]], %[[C42]], %[[C43]]) : (i32, i32, i32, i32) -> i32
|
|
%y = "test.op_commutative"(%c42_i32, %arg0, %arg1, %c43_i32) : (i32, i32, i32, i32) -> i32
|
|
|
|
// CHECK-NEXT: %[[O1:.*]] = "test.op_commutative"(%[[ARG_0]], %[[ARG_1]], %[[C42]], %[[C43]]) : (i32, i32, i32, i32) -> i32
|
|
%z = "test.op_commutative"(%arg0, %c42_i32, %c43_i32, %arg1): (i32, i32, i32, i32) -> i32
|
|
// CHECK-NEXT: return %[[O0]], %[[O1]]
|
|
return %y, %z: i32, i32
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: func @test_commutative_multi_cst
|
|
func.func @test_commutative_multi_cst(%arg0: i32, %arg1: i32) -> (i32, i32) {
|
|
// CHECK-NEXT: %c42_i32 = arith.constant 42 : i32
|
|
%c42_i32 = arith.constant 42 : i32
|
|
%c42_i32_2 = arith.constant 42 : i32
|
|
// CHECK-NEXT: %[[O0:.*]] = "test.op_commutative"(%arg0, %arg1, %c42_i32, %c42_i32) : (i32, i32, i32, i32) -> i32
|
|
%y = "test.op_commutative"(%c42_i32, %arg0, %arg1, %c42_i32_2) : (i32, i32, i32, i32) -> i32
|
|
|
|
%c42_i32_3 = arith.constant 42 : i32
|
|
|
|
// CHECK-NEXT: %[[O1:.*]] = "test.op_commutative"(%arg0, %arg1, %c42_i32, %c42_i32) : (i32, i32, i32, i32) -> i32
|
|
%z = "test.op_commutative"(%arg0, %c42_i32_3, %c42_i32_2, %arg1): (i32, i32, i32, i32) -> i32
|
|
// CHECK-NEXT: return %[[O0]], %[[O1]]
|
|
return %y, %z: i32, i32
|
|
}
|
|
|
|
// CHECK-LABEL: func @typemismatch
|
|
|
|
func.func @typemismatch() -> i32 {
|
|
%c42 = arith.constant 42.0 : f32
|
|
|
|
// The "passthrough_fold" folder will naively return its operand, but we don't
|
|
// want to fold here because of the type mismatch.
|
|
|
|
// CHECK: "test.passthrough_fold"
|
|
%0 = "test.passthrough_fold"(%c42) : (f32) -> (i32)
|
|
return %0 : i32
|
|
}
|
|
|
|
// CHECK-LABEL: test_dialect_canonicalizer
|
|
func.func @test_dialect_canonicalizer() -> (i32) {
|
|
%0 = "test.dialect_canonicalizable"() : () -> (i32)
|
|
// CHECK: %[[CST:.*]] = arith.constant 42 : i32
|
|
// CHECK: return %[[CST]]
|
|
return %0 : i32
|
|
}
|
|
|
|
// Check that the option to control region simplification actually works
|
|
// CHECK-LABEL: test_region_simplify
|
|
func.func @test_region_simplify() {
|
|
// CHECK-NEXT: return
|
|
// NO-RS-NEXT: ^bb1
|
|
// NO-RS-NEXT: return
|
|
// CHECK-NEXT: }
|
|
return
|
|
^bb1:
|
|
return
|
|
}
|