xref: /llvm-project/mlir/test/Conversion/SCFToEmitC/if.mlir (revision 977d744b21d06415ac872258bf86e026d8eb487f) 
1// RUN: mlir-opt -allow-unregistered-dialect -convert-scf-to-emitc %s | FileCheck %s
2
3func.func @test_if(%arg0: i1, %arg1: f32) {
4  scf.if %arg0 {
5     %0 = emitc.call_opaque "func_const"(%arg1) : (f32) -> i32
6  }
7  return
8}
9// CHECK-LABEL: func.func @test_if(
10// CHECK-SAME:                     %[[VAL_0:.*]]: i1,
11// CHECK-SAME:                     %[[VAL_1:.*]]: f32) {
12// CHECK-NEXT:    emitc.if %[[VAL_0]] {
13// CHECK-NEXT:      %[[VAL_2:.*]] = call_opaque "func_const"(%[[VAL_1]]) : (f32) -> i32
14// CHECK-NEXT:    }
15// CHECK-NEXT:    return
16// CHECK-NEXT:  }
17
18
19func.func @test_if_else(%arg0: i1, %arg1: f32) {
20  scf.if %arg0 {
21    %0 = emitc.call_opaque "func_true"(%arg1) : (f32) -> i32
22  } else {
23    %0 = emitc.call_opaque "func_false"(%arg1) : (f32) -> i32
24  }
25  return
26}
27// CHECK-LABEL: func.func @test_if_else(
28// CHECK-SAME:                          %[[VAL_0:.*]]: i1,
29// CHECK-SAME:                          %[[VAL_1:.*]]: f32) {
30// CHECK-NEXT:    emitc.if %[[VAL_0]] {
31// CHECK-NEXT:      %[[VAL_2:.*]] = call_opaque "func_true"(%[[VAL_1]]) : (f32) -> i32
32// CHECK-NEXT:    } else {
33// CHECK-NEXT:      %[[VAL_3:.*]] = call_opaque "func_false"(%[[VAL_1]]) : (f32) -> i32
34// CHECK-NEXT:    }
35// CHECK-NEXT:    return
36// CHECK-NEXT:  }
37
38
39func.func @test_if_yield(%arg0: i1, %arg1: f32) -> (i32, f64) {
40  %0 = arith.constant 0 : i8
41  %x, %y = scf.if %arg0 -> (i32, f64) {
42    %1 = emitc.call_opaque "func_true_1"(%arg1) : (f32) -> i32
43    %2 = emitc.call_opaque "func_true_2"(%arg1) : (f32) -> f64
44    scf.yield %1, %2 : i32, f64
45  } else {
46    %1 = emitc.call_opaque "func_false_1"(%arg1) : (f32) -> i32
47    %2 = emitc.call_opaque "func_false_2"(%arg1) : (f32) -> f64
48    scf.yield %1, %2 : i32, f64
49  }
50  return %x, %y : i32, f64
51}
52// CHECK-LABEL: func.func @test_if_yield(
53// CHECK-SAME:                           %[[VAL_0:.*]]: i1,
54// CHECK-SAME:                           %[[VAL_1:.*]]: f32) -> (i32, f64) {
55// CHECK-NEXT:    %[[VAL_2:.*]] = arith.constant 0 : i8
56// CHECK-NEXT:    %[[VAL_3:.*]] = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<i32>
57// CHECK-NEXT:    %[[VAL_4:.*]] = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<f64>
58// CHECK-NEXT:    emitc.if %[[VAL_0]] {
59// CHECK-NEXT:      %[[VAL_5:.*]] = call_opaque "func_true_1"(%[[VAL_1]]) : (f32) -> i32
60// CHECK-NEXT:      %[[VAL_6:.*]] = call_opaque "func_true_2"(%[[VAL_1]]) : (f32) -> f64
61// CHECK-NEXT:      assign %[[VAL_5]] : i32 to %[[VAL_3]] : <i32>
62// CHECK-NEXT:      assign %[[VAL_6]] : f64 to %[[VAL_4]] : <f64>
63// CHECK-NEXT:    } else {
64// CHECK-NEXT:      %[[VAL_7:.*]] = call_opaque "func_false_1"(%[[VAL_1]]) : (f32) -> i32
65// CHECK-NEXT:      %[[VAL_8:.*]] = call_opaque "func_false_2"(%[[VAL_1]]) : (f32) -> f64
66// CHECK-NEXT:      assign %[[VAL_7]] : i32 to %[[VAL_3]] : <i32>
67// CHECK-NEXT:      assign %[[VAL_8]] : f64 to %[[VAL_4]] : <f64>
68// CHECK-NEXT:    }
69// CHECK-NEXT:    %[[VAL_9:.*]] = emitc.load %[[VAL_3]] : <i32>
70// CHECK-NEXT:    %[[VAL_10:.*]] = emitc.load %[[VAL_4]] : <f64>
71// CHECK-NEXT:    return %[[VAL_9]], %[[VAL_10]] : i32, f64
72// CHECK-NEXT:  }
73