7e87d03b45
This PR should fix the parsing bug reported in https://github.com/llvm/llvm-project/issues/87430. It allows using result number as the `cf.switch` operand.
54 lines
1.1 KiB
MLIR
54 lines
1.1 KiB
MLIR
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
|
|
// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s
|
|
|
|
// CHECK-LABEL: @assert
|
|
func.func @assert(%arg : i1) {
|
|
cf.assert %arg, "Some message in case this assertion fails."
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func @switch(
|
|
func.func @switch(%flag : i32, %caseOperand : i32) {
|
|
cf.switch %flag : i32, [
|
|
default: ^bb1(%caseOperand : i32),
|
|
42: ^bb2(%caseOperand : i32),
|
|
43: ^bb3(%caseOperand : i32)
|
|
]
|
|
|
|
^bb1(%bb1arg : i32):
|
|
return
|
|
^bb2(%bb2arg : i32):
|
|
return
|
|
^bb3(%bb3arg : i32):
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func @switch_i64(
|
|
func.func @switch_i64(%flag : i64, %caseOperand : i32) {
|
|
cf.switch %flag : i64, [
|
|
default: ^bb1(%caseOperand : i32),
|
|
42: ^bb2(%caseOperand : i32),
|
|
43: ^bb3(%caseOperand : i32)
|
|
]
|
|
|
|
^bb1(%bb1arg : i32):
|
|
return
|
|
^bb2(%bb2arg : i32):
|
|
return
|
|
^bb3(%bb3arg : i32):
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func @switch_result_number
|
|
func.func @switch_result_number(%arg0: i32) {
|
|
%0:2 = "test.op_with_two_results"() : () -> (i32, i32)
|
|
cf.switch %arg0 : i32, [
|
|
default: ^bb2,
|
|
0: ^bb1(%0#0 : i32)
|
|
]
|
|
^bb1(%1: i32):
|
|
return
|
|
^bb2:
|
|
return
|
|
}
|