103 lines · plain
1// RUN: mlir-opt -allow-unregistered-dialect -test-scf-if-utils -split-input-file %s | FileCheck %s2 3// -----4 5// CHECK: func @outlined_then0(%{{.*}}: i1, %{{.*}}: memref<?xf32>) -> i8 {6// CHECK-NEXT: %{{.*}} = "some_op"(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> i87// CHECK-NEXT: return %{{.*}} : i88// CHECK-NEXT: }9// CHECK: func @outlined_else0(%{{.*}}: i8) -> i8 {10// CHECK-NEXT: return %{{.*}}0 : i811// CHECK-NEXT: }12// CHECK: func @outline_if_else(13// CHECK-NEXT: %{{.*}} = scf.if %{{.*}} -> (i8) {14// CHECK-NEXT: %{{.*}} = func.call @outlined_then0(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> i815// CHECK-NEXT: scf.yield %{{.*}} : i816// CHECK-NEXT: } else {17// CHECK-NEXT: %{{.*}} = func.call @outlined_else0(%{{.*}}) : (i8) -> i818// CHECK-NEXT: scf.yield %{{.*}} : i819// CHECK-NEXT: }20// CHECK-NEXT: return21// CHECK-NEXT: }22func.func @outline_if_else(%cond: i1, %a: index, %b: memref<?xf32>, %c: i8) {23 %r = scf.if %cond -> (i8) {24 %r = "some_op"(%cond, %b) : (i1, memref<?xf32>) -> (i8)25 scf.yield %r : i826 } else {27 scf.yield %c : i828 }29 return30}31 32// -----33 34// CHECK: func @outlined_then0(%{{.*}}: i1, %{{.*}}: memref<?xf32>) {35// CHECK-NEXT: "some_op"(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> ()36// CHECK-NEXT: return37// CHECK-NEXT: }38// CHECK: func @outline_if(39// CHECK-NEXT: scf.if %{{.*}} {40// CHECK-NEXT: func.call @outlined_then0(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> ()41// CHECK-NEXT: }42// CHECK-NEXT: return43// CHECK-NEXT: }44func.func @outline_if(%cond: i1, %a: index, %b: memref<?xf32>, %c: i8) {45 scf.if %cond {46 "some_op"(%cond, %b) : (i1, memref<?xf32>) -> ()47 scf.yield48 }49 return50}51 52// -----53 54// CHECK: func @outlined_then0() {55// CHECK-NEXT: return56// CHECK-NEXT: }57// CHECK: func @outlined_else0(%{{.*}}: i1, %{{.*}}: memref<?xf32>) {58// CHECK-NEXT: "some_op"(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> ()59// CHECK-NEXT: return60// CHECK-NEXT: }61// CHECK: func @outline_empty_if_else(62// CHECK-NEXT: scf.if %{{.*}} {63// CHECK-NEXT: func.call @outlined_then0() : () -> ()64// CHECK-NEXT: } else {65// CHECK-NEXT: func.call @outlined_else0(%{{.*}}, %{{.*}}) : (i1, memref<?xf32>) -> ()66// CHECK-NEXT: }67// CHECK-NEXT: return68// CHECK-NEXT: }69func.func @outline_empty_if_else(%cond: i1, %a: index, %b: memref<?xf32>, %c: i8) {70 scf.if %cond {71 } else {72 "some_op"(%cond, %b) : (i1, memref<?xf32>) -> ()73 }74 return75}76 77// -----78 79// This test checks scf utils can work on llvm func.80 81// CHECK: func @outlined_then0() {82// CHECK-NEXT: return83// CHECK-NEXT: }84// CHECK: func @outlined_else0(%{{.*}}: i1, %{{.*}}: i32) {85// CHECK-NEXT: "some_op"(%{{.*}}, %{{.*}}) : (i1, i32) -> ()86// CHECK-NEXT: return87// CHECK-NEXT: }88// CHECK: llvm.func @llvm_func(%{{.*}}: i1, %{{.*}}: i32) {89// CHECK-NEXT: scf.if %{{.*}} {90// CHECK-NEXT: func.call @outlined_then0() : () -> ()91// CHECK-NEXT: } else {92// CHECK-NEXT: func.call @outlined_else0(%{{.*}}, %{{.*}}) : (i1, i32) -> ()93// CHECK-NEXT: }94// CHECK-NEXT: llvm.return95// CHECK-NEXT: }96llvm.func @llvm_func(%cond: i1, %a: i32) {97 scf.if %cond {98 } else {99 "some_op"(%cond, %a) : (i1, i32) -> ()100 }101 llvm.return102}103