211 lines · plain
1// RUN: mlir-opt %s -form-expressions | FileCheck %s2 3// CHECK-LABEL: func.func @single_expression(4// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: i32, %[[VAL_3:.*]]: i32) -> i1 {5// CHECK: %[[VAL_5:.*]] = emitc.expression %[[VAL_3]], %[[VAL_2]], %[[VAL_0]] : (i32, i32, i32) -> i1 {6// CHECK: %[[VAL_4:.*]] = "emitc.constant"() <{value = 42 : i32}> : () -> i327// CHECK: %[[VAL_6:.*]] = mul %[[VAL_0]], %[[VAL_4]] : (i32, i32) -> i328// CHECK: %[[VAL_7:.*]] = sub %[[VAL_6]], %[[VAL_2]] : (i32, i32) -> i329// CHECK: %[[VAL_8:.*]] = cmp lt, %[[VAL_7]], %[[VAL_3]] : (i32, i32) -> i110// CHECK: yield %[[VAL_8]] : i111// CHECK: }12// CHECK: return %[[VAL_5]] : i113// CHECK: }14 15func.func @single_expression(%arg0: i32, %arg1: i32, %arg2: i32, %arg3: i32) -> i1 {16 %c42 = "emitc.constant"(){value = 42 : i32} : () -> i3217 %a = emitc.mul %arg0, %c42 : (i32, i32) -> i3218 %b = emitc.sub %a, %arg2 : (i32, i32) -> i3219 %c = emitc.cmp lt, %b, %arg3 :(i32, i32) -> i120 return %c : i121}22 23// CHECK-LABEL: func.func @multiple_expressions(24// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: i32, %[[VAL_3:.*]]: i32) -> (i32, i32) {25// CHECK: %[[VAL_4:.*]] = emitc.expression %[[VAL_2]], %[[VAL_0]], %[[VAL_1]] : (i32, i32, i32) -> i32 {26// CHECK: %[[VAL_5:.*]] = mul %[[VAL_0]], %[[VAL_1]] : (i32, i32) -> i3227// CHECK: %[[VAL_6:.*]] = sub %[[VAL_5]], %[[VAL_2]] : (i32, i32) -> i3228// CHECK: yield %[[VAL_6]] : i3229// CHECK: }30// CHECK: %[[VAL_7:.*]] = emitc.expression %[[VAL_2]], %[[VAL_1]], %[[VAL_3]] : (i32, i32, i32) -> i32 {31// CHECK: %[[VAL_8:.*]] = add %[[VAL_1]], %[[VAL_3]] : (i32, i32) -> i3232// CHECK: %[[VAL_9:.*]] = div %[[VAL_8]], %[[VAL_2]] : (i32, i32) -> i3233// CHECK: yield %[[VAL_9]] : i3234// CHECK: }35// CHECK: return %[[VAL_4]], %[[VAL_7]] : i32, i3236// CHECK: }37 38func.func @multiple_expressions(%arg0: i32, %arg1: i32, %arg2: i32, %arg3: i32) -> (i32, i32) {39 %a = emitc.mul %arg0, %arg1 : (i32, i32) -> i3240 %b = emitc.sub %a, %arg2 : (i32, i32) -> i3241 %c = emitc.add %arg1, %arg3 : (i32, i32) -> i3242 %d = emitc.div %c, %arg2 : (i32, i32) -> i3243 return %b, %d : i32, i3244}45 46// CHECK-LABEL: func.func @expression_with_call(47// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: i32, %[[VAL_3:.*]]: i32) -> i1 {48// CHECK: %[[VAL_4:.*]] = emitc.expression %[[VAL_2]], %[[VAL_0]], %[[VAL_1]] : (i32, i32, i32) -> i32 {49// CHECK: %[[VAL_5:.*]] = mul %[[VAL_0]], %[[VAL_1]] : (i32, i32) -> i3250// CHECK: %[[VAL_6:.*]] = call_opaque "foo"(%[[VAL_5]], %[[VAL_2]]) : (i32, i32) -> i3251// CHECK: yield %[[VAL_6]] : i3252// CHECK: }53// CHECK: %[[VAL_7:.*]] = emitc.expression %[[VAL_4]], %[[VAL_1]] : (i32, i32) -> i1 {54// CHECK: %[[VAL_8:.*]] = cmp lt, %[[VAL_4]], %[[VAL_1]] : (i32, i32) -> i155// CHECK: yield %[[VAL_8]] : i156// CHECK: }57// CHECK: return %[[VAL_7]] : i158// CHECK: }59 60func.func @expression_with_call(%arg0: i32, %arg1: i32, %arg2: i32, %arg3: i32) -> i1 {61 %a = emitc.mul %arg0, %arg1 : (i32, i32) -> i3262 %b = emitc.call_opaque "foo" (%a, %arg2) : (i32, i32) -> (i32)63 %c = emitc.cmp lt, %b, %arg1 :(i32, i32) -> i164 return %c : i165}66 67// CHECK-LABEL: func.func @expression_with_dereference(68// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: !emitc.ptr<i32>) -> i1 {69// CHECK: %[[VAL_3:.*]] = emitc.expression %[[VAL_2]] : (!emitc.ptr<i32>) -> i32 {70// CHECK: %[[VAL_4:.*]] = apply "*"(%[[VAL_2]]) : (!emitc.ptr<i32>) -> i3271// CHECK: yield %[[VAL_4]] : i3272// CHECK: }73// CHECK: %[[VAL_5:.*]] = emitc.expression %[[VAL_3]], %[[VAL_0]], %[[VAL_1]] : (i32, i32, i32) -> i1 {74// CHECK: %[[VAL_6:.*]] = mul %[[VAL_0]], %[[VAL_1]] : (i32, i32) -> i3275// CHECK: %[[VAL_7:.*]] = cmp lt, %[[VAL_6]], %[[VAL_3]] : (i32, i32) -> i176// CHECK: return %[[VAL_5]] : i177// CHECK: }78 79func.func @expression_with_dereference(%arg0: i32, %arg1: i32, %arg2: !emitc.ptr<i32>) -> i1 {80 %a = emitc.mul %arg0, %arg1 : (i32, i32) -> i3281 %b = emitc.apply "*"(%arg2) : (!emitc.ptr<i32>) -> (i32)82 %c = emitc.cmp lt, %a, %b :(i32, i32) -> i183 return %c : i184}85 86// CHECK-LABEL: func.func @expression_with_address_taken(87// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: !emitc.ptr<i32>) -> i1 {88// CHECK: %[[VAL_3:.*]] = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<i32>89// CHECK: %[[VAL_4:.*]] = emitc.expression %[[VAL_2]], %[[VAL_1]], %[[VAL_3]] : (!emitc.ptr<i32>, i32, !emitc.lvalue<i32>) -> i1 {90// CHECK: %[[VAL_5:.*]] = apply "&"(%[[VAL_3]]) : (!emitc.lvalue<i32>) -> !emitc.ptr<i32>91// CHECK: %[[VAL_6:.*]] = add %[[VAL_5]], %[[VAL_1]] : (!emitc.ptr<i32>, i32) -> !emitc.ptr<i32>92// CHECK: %[[VAL_7:.*]] = cmp lt, %[[VAL_6]], %[[VAL_2]] : (!emitc.ptr<i32>, !emitc.ptr<i32>) -> i193// CHECK: yield %[[VAL_7]] : i194// CHECK: }95// CHECK: return %[[VAL_4]] : i196// CHECK: }97 98func.func @expression_with_address_taken(%arg0: i32, %arg1: i32, %arg2: !emitc.ptr<i32>) -> i1 {99 %0 = "emitc.variable"() <{value = #emitc.opaque<"">}> : () -> !emitc.lvalue<i32>100 %a = emitc.apply "&"(%0) : (!emitc.lvalue<i32>) -> !emitc.ptr<i32>101 %b = emitc.add %a, %arg1 : (!emitc.ptr<i32>, i32) -> !emitc.ptr<i32>102 %c = emitc.cmp lt, %b, %arg2 :(!emitc.ptr<i32>, !emitc.ptr<i32>) -> i1103 return %c : i1104}105 106// CHECK-LABEL: func.func @no_nested_expression(107// CHECK-SAME: %[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32) -> i1 {108// CHECK: %[[VAL_2:.*]] = emitc.expression %[[VAL_0]], %[[VAL_1]] : (i32, i32) -> i1 {109// CHECK: %[[VAL_3:.*]] = cmp lt, %[[VAL_0]], %[[VAL_1]] : (i32, i32) -> i1110// CHECK: yield %[[VAL_3]] : i1111// CHECK: }112// CHECK: return %[[VAL_2]] : i1113// CHECK: }114 115func.func @no_nested_expression(%arg0: i32, %arg1: i32) -> i1 {116 %a = emitc.expression %arg0, %arg1 :(i32, i32) -> i1 {117 %b = emitc.cmp lt, %arg0, %arg1 :(i32, i32) -> i1118 emitc.yield %b : i1119 }120 return %a : i1121}122 123// CHECK-LABEL: func.func @single_result_requirement124// CHECK-NOT: emitc.expression125 126func.func @single_result_requirement() -> (i32, i32) {127 %0:2 = emitc.call_opaque "foo" () : () -> (i32, i32)128 return %0#0, %0#1 : i32, i32129}130 131// CHECK-LABEL: func.func @expression_with_load(132// CHECK-SAME: %[[VAL_0:.*]]: i32,133// CHECK-SAME: %[[VAL_1:.*]]: !emitc.ptr<i32>) -> i1 {134// CHECK: %[[VAL_2:.*]] = emitc.expression : () -> i64 {135// CHECK: %[[VAL_C:.*]] = "emitc.constant"() <{value = 0 : i64}> : () -> i64136// CHECK: yield %[[VAL_C]] : i64137// CHECK: }138// CHECK: %[[VAL_3:.*]] = "emitc.variable"() <{value = #emitc.opaque<"42">}> : () -> !emitc.lvalue<i32>139// CHECK: %[[VAL_4:.*]] = emitc.expression %[[VAL_3]] : (!emitc.lvalue<i32>) -> i32 {140// CHECK: %[[VAL_5:.*]] = load %[[VAL_3]] : <i32>141// CHECK: yield %[[VAL_5]] : i32142// CHECK: }143// CHECK: %[[VAL_6:.*]] = emitc.subscript %[[VAL_1]]{{\[}}%[[VAL_2]]] : (!emitc.ptr<i32>, i64) -> !emitc.lvalue<i32>144// CHECK: %[[VAL_7:.*]] = emitc.expression %[[VAL_6]] : (!emitc.lvalue<i32>) -> i32 {145// CHECK: %[[VAL_8:.*]] = load %[[VAL_6]] : <i32>146// CHECK: yield %[[VAL_8]] : i32147// CHECK: }148// CHECK: %[[VAL_9:.*]] = emitc.expression %[[VAL_0]], %[[VAL_4]], %[[VAL_7]] : (i32, i32, i32) -> i1 {149// CHECK: %[[VAL_10:.*]] = add %[[VAL_4]], %[[VAL_7]] : (i32, i32) -> i32150// CHECK: %[[VAL_11:.*]] = cmp lt, %[[VAL_10]], %[[VAL_0]] : (i32, i32) -> i1151// CHECK: yield %[[VAL_11]] : i1152// CHECK: }153// CHECK: return %[[VAL_9]] : i1154// CHECK: }155 156func.func @expression_with_load(%arg0: i32, %arg1: !emitc.ptr<i32>) -> i1 {157 %c0 = "emitc.constant"() {value = 0 : i64} : () -> i64158 %0 = "emitc.variable"() <{value = #emitc.opaque<"42">}> : () -> !emitc.lvalue<i32>159 %a = emitc.load %0 : !emitc.lvalue<i32>160 %ptr = emitc.subscript %arg1[%c0] : (!emitc.ptr<i32>, i64) -> !emitc.lvalue<i32>161 %ptr_load = emitc.load %ptr : !emitc.lvalue<i32>162 %b = emitc.add %a, %ptr_load : (i32, i32) -> i32163 %c = emitc.cmp lt, %b, %arg0 :(i32, i32) -> i1164 return %c : i1165}166 167// CHECK-LABEL: func.func @opaque_type_expression(%arg0: i32, %arg1: !emitc.opaque<"T0">, %arg2: i32) -> i1 {168// CHECK: %0 = emitc.expression : () -> !emitc.opaque<"T1"> {169// CHECK: %4 = "emitc.constant"() <{value = #emitc.opaque<"V">}> : () -> !emitc.opaque<"T1">170// CHECK: yield %4 : !emitc.opaque<"T1">171// CHECK: }172// CHECK: %1 = emitc.expression %arg0, %0 : (i32, !emitc.opaque<"T1">) -> i32 {173// CHECK: %4 = mul %arg0, %0 : (i32, !emitc.opaque<"T1">) -> i32174// CHECK: yield %4 : i32175// CHECK: }176// CHECK: %2 = emitc.expression %1, %arg1 : (i32, !emitc.opaque<"T0">) -> i32 {177// CHECK: %4 = sub %1, %arg1 : (i32, !emitc.opaque<"T0">) -> i32178// CHECK: yield %4 : i32179// CHECK: }180// CHECK: %3 = emitc.expression %2, %arg2 : (i32, i32) -> i1 {181// CHECK: %4 = cmp lt, %2, %arg2 : (i32, i32) -> i1182// CHECK: yield %4 : i1183// CHECK: }184// CHECK: return %3 : i1185// CHECK: }186 187 188func.func @opaque_type_expression(%arg0: i32, %arg1: !emitc.opaque<"T0">, %arg2: i32) -> i1 {189 %c42 = "emitc.constant"(){value = #emitc.opaque<"V">} : () -> !emitc.opaque<"T1">190 %a = emitc.mul %arg0, %c42 : (i32, !emitc.opaque<"T1">) -> i32191 %b = emitc.sub %a, %arg1 : (i32, !emitc.opaque<"T0">) -> i32192 %c = emitc.cmp lt, %b, %arg2 :(i32, i32) -> i1193 return %c : i1194}195 196// CHECK-LABEL: func.func @expression_with_constant(197// CHECK-SAME: %[[VAL_0:.*]]: i32) -> i32 {198// CHECK: %[[VAL_1:.*]] = emitc.expression %[[VAL_0]] : (i32) -> i32 {199// CHECK: %[[VAL_2:.*]] = "emitc.constant"() <{value = 42 : i32}> : () -> i32200// CHECK: %[[VAL_3:.*]] = mul %[[VAL_0]], %[[VAL_2]] : (i32, i32) -> i32201// CHECK: yield %[[VAL_3]] : i32202// CHECK: }203// CHECK: return %[[VAL_1]] : i32204// CHECK: }205 206func.func @expression_with_constant(%arg0: i32) -> i32 {207 %c42 = "emitc.constant"(){value = 42 : i32} : () -> i32208 %a = emitc.mul %arg0, %c42 : (i32, i32) -> i32209 return %a : i32210}211