74 lines · plain
1// RUN: mlir-opt -allow-unregistered-dialect -convert-scf-to-emitc %s | FileCheck %s2// RUN: mlir-opt -allow-unregistered-dialect -convert-to-emitc="filter-dialects=scf" %s | FileCheck %s3 4func.func @test_if(%arg0: i1, %arg1: f32) {5 scf.if %arg0 {6 %0 = emitc.call_opaque "func_const"(%arg1) : (f32) -> i327 }8 return9}10// CHECK-LABEL: func.func @test_if(11// CHECK-SAME: %[[VAL_0:.*]]: i1,12// CHECK-SAME: %[[VAL_1:.*]]: f32) {13// CHECK-NEXT: emitc.if %[[VAL_0]] {14// CHECK-NEXT: %[[VAL_2:.*]] = call_opaque "func_const"(%[[VAL_1]]) : (f32) -> i3215// CHECK-NEXT: }16// CHECK-NEXT: return17// CHECK-NEXT: }18 19 20func.func @test_if_else(%arg0: i1, %arg1: f32) {21 scf.if %arg0 {22 %0 = emitc.call_opaque "func_true"(%arg1) : (f32) -> i3223 } else {24 %0 = emitc.call_opaque "func_false"(%arg1) : (f32) -> i3225 }26 return27}28// CHECK-LABEL: func.func @test_if_else(29// CHECK-SAME: %[[VAL_0:.*]]: i1,30// CHECK-SAME: %[[VAL_1:.*]]: f32) {31// CHECK-NEXT: emitc.if %[[VAL_0]] {32// CHECK-NEXT: %[[VAL_2:.*]] = call_opaque "func_true"(%[[VAL_1]]) : (f32) -> i3233// CHECK-NEXT: } else {34// CHECK-NEXT: %[[VAL_3:.*]] = call_opaque "func_false"(%[[VAL_1]]) : (f32) -> i3235// CHECK-NEXT: }36// CHECK-NEXT: return37// CHECK-NEXT: }38 39 40func.func @test_if_yield(%arg0: i1, %arg1: f32) -> (i32, f64) {41 %0 = arith.constant 0 : i842 %x, %y = scf.if %arg0 -> (i32, f64) {43 %1 = emitc.call_opaque "func_true_1"(%arg1) : (f32) -> i3244 %2 = emitc.call_opaque "func_true_2"(%arg1) : (f32) -> f6445 scf.yield %1, %2 : i32, f6446 } else {47 %1 = emitc.call_opaque "func_false_1"(%arg1) : (f32) -> i3248 %2 = emitc.call_opaque "func_false_2"(%arg1) : (f32) -> f6449 scf.yield %1, %2 : i32, f6450 }51 return %x, %y : i32, f6452}53// CHECK-LABEL: func.func @test_if_yield(54// CHECK-SAME: %[[VAL_0:.*]]: i1,55// CHECK-SAME: %[[VAL_1:.*]]: f32) -> (i32, f64) {56// CHECK-NEXT: %[[VAL_2:.*]] = arith.constant 0 : i857// CHECK-NEXT: %[[VAL_3:.*]] = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<i32>58// CHECK-NEXT: %[[VAL_4:.*]] = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<f64>59// CHECK-NEXT: emitc.if %[[VAL_0]] {60// CHECK-NEXT: %[[VAL_5:.*]] = call_opaque "func_true_1"(%[[VAL_1]]) : (f32) -> i3261// CHECK-NEXT: %[[VAL_6:.*]] = call_opaque "func_true_2"(%[[VAL_1]]) : (f32) -> f6462// CHECK-NEXT: assign %[[VAL_5]] : i32 to %[[VAL_3]] : <i32>63// CHECK-NEXT: assign %[[VAL_6]] : f64 to %[[VAL_4]] : <f64>64// CHECK-NEXT: } else {65// CHECK-NEXT: %[[VAL_7:.*]] = call_opaque "func_false_1"(%[[VAL_1]]) : (f32) -> i3266// CHECK-NEXT: %[[VAL_8:.*]] = call_opaque "func_false_2"(%[[VAL_1]]) : (f32) -> f6467// CHECK-NEXT: assign %[[VAL_7]] : i32 to %[[VAL_3]] : <i32>68// CHECK-NEXT: assign %[[VAL_8]] : f64 to %[[VAL_4]] : <f64>69// CHECK-NEXT: }70// CHECK-NEXT: %[[VAL_9:.*]] = emitc.load %[[VAL_3]] : <i32>71// CHECK-NEXT: %[[VAL_10:.*]] = emitc.load %[[VAL_4]] : <f64>72// CHECK-NEXT: return %[[VAL_9]], %[[VAL_10]] : i32, f6473// CHECK-NEXT: }74