40 lines · plain
1// Check that the test version of the wide integer emulation pass applies2// conversion to functions whose name start with a given prefix only, and that3// the function signatures are preserved.4 5// RUN: mlir-opt %s --test-arith-emulate-wide-int="function-prefix=emulate_me_" | FileCheck %s6 7// CHECK-LABEL: func.func @entry()8// CHECK: {{%.+}} = call @emulate_me_please({{.+}}) : (i64) -> i649// CHECK-NEXT: {{%.+}} = call @foo({{.+}}) : (i64) -> i6410func.func @entry() {11 %cst0 = arith.constant 0 : i6412 func.call @emulate_me_please(%cst0) : (i64) -> (i64)13 func.call @foo(%cst0) : (i64) -> (i64)14 return15}16 17// CHECK-LABEL: func.func @emulate_me_please18// CHECK-SAME: ([[ARG:%.+]]: i64) -> i64 {19// CHECK-NEXT: [[BCAST0:%.+]] = llvm.bitcast [[ARG]] : i64 to vector<2xi32>20// CHECK-NEXT: [[LOW0:%.+]] = vector.extract [[BCAST0]][0] : i32 from vector<2xi32>21// CHECK-NEXT: [[HIGH0:%.+]] = vector.extract [[BCAST0]][1] : i32 from vector<2xi32>22// CHECK-NEXT: [[LOW1:%.+]] = vector.extract [[BCAST0]][0] : i32 from vector<2xi32>23// CHECK-NEXT: [[HIGH1:%.+]] = vector.extract [[BCAST0]][1] : i32 from vector<2xi32>24// CHECK-NEXT: {{%.+}}, {{%.+}} = arith.addui_extended [[LOW0]], [[LOW1]] : i32, i125// CHECK: [[RES:%.+]] = llvm.bitcast {{%.+}} : vector<2xi32> to i6426// CHECK-NEXT: return [[RES]] : i6427func.func @emulate_me_please(%x : i64) -> i64 {28 %r = arith.addi %x, %x : i6429 return %r : i6430}31 32// CHECK-LABEL: func.func @foo33// CHECK-SAME: ([[ARG:%.+]]: i64) -> i64 {34// CHECK-NEXT: [[RES:%.+]] = arith.addi [[ARG]], [[ARG]] : i6435// CHECK-NEXT: return [[RES]] : i6436func.func @foo(%x : i64) -> i64 {37 %r = arith.addi %x, %x : i6438 return %r : i6439}40