0f8a6b7d03
This is intended as a fast pattern rewrite driver for the cases when a simple walk gets the job done but we would still want to implement it in terms of rewrite patterns (that can be used with the greedy pattern rewrite driver downstream). The new driver is inspired by the discussion in https://github.com/llvm/llvm-project/pull/112454 and the LLVM Dev presentation from @matthias-springer earlier this week. This limitation comes with some limitations: * It does not repeat until a fixpoint or revisit ops modified in place or newly created ops. In general, it only walks forward (in the post-order). * `matchAndRewrite` can only erase the matched op or its descendants. This is verified under expensive checks. * It does not perform folding / DCE. We could probably relax some of these in the future without sacrificing too much performance.
36 lines
1.2 KiB
MLIR
36 lines
1.2 KiB
MLIR
// RUN: mlir-opt %s -test-greedy-patterns="max-iterations=1" \
|
|
// RUN: -allow-unregistered-dialect --split-input-file | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @add_to_worklist_after_inplace_update()
|
|
func.func @add_to_worklist_after_inplace_update() {
|
|
// The following op is updated in-place and should be added back to the
|
|
// worklist of the GreedyPatternRewriteDriver (regardless of the value of
|
|
// config.max_iterations).
|
|
|
|
// CHECK: "test.any_attr_of_i32_str"() <{attr = 3 : i32}> : () -> ()
|
|
"test.any_attr_of_i32_str"() {attr = 0 : i32} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @add_ancestors_to_worklist()
|
|
func.func @add_ancestors_to_worklist() {
|
|
// CHECK: "foo.maybe_eligible_op"() {eligible} : () -> index
|
|
// CHECK-NEXT: "test.one_region_op"()
|
|
"test.one_region_op"() ({
|
|
%0 = "foo.maybe_eligible_op" () : () -> (index)
|
|
"foo.yield"(%0) : (index) -> ()
|
|
}) {hoist_eligible_ops}: () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// There are no patterns in this test that apply to "test.symbol". This test is
|
|
// to ensure that symbols are not getting removed due to being "trivially dead"
|
|
// as part of a greedy rewrite. Symbols are never trivially dead.
|
|
|
|
// CHECK: "test.symbol"() <{sym_name = "foo"}>
|
|
"test.symbol"() <{sym_name = "foo"}> : () -> ()
|