3655069234
This commit moves FuncOp out of the builtin dialect, and into the Func dialect. This move has been planned in some capacity from the moment we made FuncOp an operation (years ago). This commit handles the functional aspects of the move, but various aspects are left untouched to ease migration: func::FuncOp is re-exported into mlir to reduce the actual API churn, the assembly format still accepts the unqualified `func`. These temporary measures will remain for a little while to simplify migration before being removed. Differential Revision: https://reviews.llvm.org/D121266
213 lines
8.5 KiB
MLIR
213 lines
8.5 KiB
MLIR
// RUN: mlir-opt -test-ir-visitors -allow-unregistered-dialect -split-input-file %s | FileCheck %s
|
|
|
|
// Verify the different configurations of IR visitors.
|
|
// Constant, yield and other terminator ops are not matched for simplicity.
|
|
// Module and function op and their immediately nested blocks are not erased in
|
|
// callbacks with return so that the output includes more cases in pre-order.
|
|
|
|
func @structured_cfg() {
|
|
%c0 = arith.constant 0 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c10 = arith.constant 10 : index
|
|
scf.for %i = %c1 to %c10 step %c1 {
|
|
%cond = "use0"(%i) : (index) -> (i1)
|
|
scf.if %cond {
|
|
"use1"(%i) : (index) -> ()
|
|
} else {
|
|
"use2"(%i) : (index) -> ()
|
|
}
|
|
"use3"(%i) : (index) -> ()
|
|
}
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: Op pre-order visit
|
|
// CHECK: Visiting op 'builtin.module'
|
|
// CHECK: Visiting op 'func.func'
|
|
// CHECK: Visiting op 'scf.for'
|
|
// CHECK: Visiting op 'use0'
|
|
// CHECK: Visiting op 'scf.if'
|
|
// CHECK: Visiting op 'use1'
|
|
// CHECK: Visiting op 'use2'
|
|
// CHECK: Visiting op 'use3'
|
|
// CHECK: Visiting op 'func.return'
|
|
|
|
// CHECK-LABEL: Block pre-order visits
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'builtin.module'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'scf.for'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'scf.if'
|
|
// CHECK: Visiting block ^bb0 from region 1 from operation 'scf.if'
|
|
|
|
// CHECK-LABEL: Region pre-order visits
|
|
// CHECK: Visiting region 0 from operation 'builtin.module'
|
|
// CHECK: Visiting region 0 from operation 'func.func'
|
|
// CHECK: Visiting region 0 from operation 'scf.for'
|
|
// CHECK: Visiting region 0 from operation 'scf.if'
|
|
// CHECK: Visiting region 1 from operation 'scf.if'
|
|
|
|
// CHECK-LABEL: Op post-order visits
|
|
// CHECK: Visiting op 'use0'
|
|
// CHECK: Visiting op 'use1'
|
|
// CHECK: Visiting op 'use2'
|
|
// CHECK: Visiting op 'scf.if'
|
|
// CHECK: Visiting op 'use3'
|
|
// CHECK: Visiting op 'scf.for'
|
|
// CHECK: Visiting op 'func.return'
|
|
// CHECK: Visiting op 'func.func'
|
|
// CHECK: Visiting op 'builtin.module'
|
|
|
|
// CHECK-LABEL: Block post-order visits
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'scf.if'
|
|
// CHECK: Visiting block ^bb0 from region 1 from operation 'scf.if'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'scf.for'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'builtin.module'
|
|
|
|
// CHECK-LABEL: Region post-order visits
|
|
// CHECK: Visiting region 0 from operation 'scf.if'
|
|
// CHECK: Visiting region 1 from operation 'scf.if'
|
|
// CHECK: Visiting region 0 from operation 'scf.for'
|
|
// CHECK: Visiting region 0 from operation 'func.func'
|
|
// CHECK: Visiting region 0 from operation 'builtin.module'
|
|
|
|
// CHECK-LABEL: Op pre-order erasures
|
|
// CHECK: Erasing op 'scf.for'
|
|
// CHECK: Erasing op 'func.return'
|
|
|
|
// CHECK-LABEL: Block pre-order erasures
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'scf.for'
|
|
|
|
// CHECK-LABEL: Op post-order erasures (skip)
|
|
// CHECK: Erasing op 'use0'
|
|
// CHECK: Erasing op 'use1'
|
|
// CHECK: Erasing op 'use2'
|
|
// CHECK: Erasing op 'scf.if'
|
|
// CHECK: Erasing op 'use3'
|
|
// CHECK: Erasing op 'scf.for'
|
|
// CHECK: Erasing op 'func.return'
|
|
|
|
// CHECK-LABEL: Block post-order erasures (skip)
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'scf.if'
|
|
// CHECK: Erasing block ^bb0 from region 1 from operation 'scf.if'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'scf.for'
|
|
|
|
// CHECK-LABEL: Op post-order erasures (no skip)
|
|
// CHECK: Erasing op 'use0'
|
|
// CHECK: Erasing op 'use1'
|
|
// CHECK: Erasing op 'use2'
|
|
// CHECK: Erasing op 'scf.if'
|
|
// CHECK: Erasing op 'use3'
|
|
// CHECK: Erasing op 'scf.for'
|
|
// CHECK: Erasing op 'func.return'
|
|
// CHECK: Erasing op 'func.func'
|
|
// CHECK: Erasing op 'builtin.module'
|
|
|
|
// CHECK-LABEL: Block post-order erasures (no skip)
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'scf.if'
|
|
// CHECK: Erasing block ^bb0 from region 1 from operation 'scf.if'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'scf.for'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'builtin.module'
|
|
|
|
// -----
|
|
|
|
func @unstructured_cfg() {
|
|
"regionOp0"() ({
|
|
^bb0:
|
|
"op0"() : () -> ()
|
|
cf.br ^bb2
|
|
^bb1:
|
|
"op1"() : () -> ()
|
|
cf.br ^bb2
|
|
^bb2:
|
|
"op2"() : () -> ()
|
|
}) : () -> ()
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: Op pre-order visits
|
|
// CHECK: Visiting op 'builtin.module'
|
|
// CHECK: Visiting op 'func.func'
|
|
// CHECK: Visiting op 'regionOp0'
|
|
// CHECK: Visiting op 'op0'
|
|
// CHECK: Visiting op 'cf.br'
|
|
// CHECK: Visiting op 'op1'
|
|
// CHECK: Visiting op 'cf.br'
|
|
// CHECK: Visiting op 'op2'
|
|
// CHECK: Visiting op 'func.return'
|
|
|
|
// CHECK-LABEL: Block pre-order visits
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'builtin.module'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting block ^bb1 from region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting block ^bb2 from region 0 from operation 'regionOp0'
|
|
|
|
// CHECK-LABEL: Region pre-order visits
|
|
// CHECK: Visiting region 0 from operation 'builtin.module'
|
|
// CHECK: Visiting region 0 from operation 'func.func'
|
|
// CHECK: Visiting region 0 from operation 'regionOp0'
|
|
|
|
// CHECK-LABEL: Op post-order visits
|
|
// CHECK: Visiting op 'op0'
|
|
// CHECK: Visiting op 'cf.br'
|
|
// CHECK: Visiting op 'op1'
|
|
// CHECK: Visiting op 'cf.br'
|
|
// CHECK: Visiting op 'op2'
|
|
// CHECK: Visiting op 'regionOp0'
|
|
// CHECK: Visiting op 'func.return'
|
|
// CHECK: Visiting op 'func.func'
|
|
// CHECK: Visiting op 'builtin.module'
|
|
|
|
// CHECK-LABEL: Block post-order visits
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting block ^bb1 from region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting block ^bb2 from region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Visiting block ^bb0 from region 0 from operation 'builtin.module'
|
|
|
|
// CHECK-LABEL: Region post-order visits
|
|
// CHECK: Visiting region 0 from operation 'regionOp0'
|
|
// CHECK: Visiting region 0 from operation 'func.func'
|
|
// CHECK: Visiting region 0 from operation 'builtin.module'
|
|
|
|
// CHECK-LABEL: Op pre-order erasures (skip)
|
|
// CHECK: Erasing op 'regionOp0'
|
|
// CHECK: Erasing op 'func.return'
|
|
|
|
// CHECK-LABEL: Block pre-order erasures (skip)
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
|
|
// CHECK-LABEL: Op post-order erasures (skip)
|
|
// CHECK: Erasing op 'op0'
|
|
// CHECK: Erasing op 'cf.br'
|
|
// CHECK: Erasing op 'op1'
|
|
// CHECK: Erasing op 'cf.br'
|
|
// CHECK: Erasing op 'op2'
|
|
// CHECK: Erasing op 'regionOp0'
|
|
// CHECK: Erasing op 'func.return'
|
|
|
|
// CHECK-LABEL: Block post-order erasures (skip)
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
|
|
// CHECK-LABEL: Op post-order erasures (no skip)
|
|
// CHECK: Erasing op 'op0'
|
|
// CHECK: Erasing op 'cf.br'
|
|
// CHECK: Erasing op 'op1'
|
|
// CHECK: Erasing op 'cf.br'
|
|
// CHECK: Erasing op 'op2'
|
|
// CHECK: Erasing op 'regionOp0'
|
|
// CHECK: Erasing op 'func.return'
|
|
|
|
// CHECK-LABEL: Block post-order erasures (no skip)
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'regionOp0'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'func.func'
|
|
// CHECK: Erasing block ^bb0 from region 0 from operation 'builtin.module'
|