mlir/test/Dialect/Arith/test-emulate-wide-int-pass.mlir

   1 // Check that the test version of the wide integer emulation pass applies
   2 // conversion to functions whose name start with a given prefix only, and that
   3 // the function signatures are preserved.
   4
   5 // RUN: mlir-opt %s --test-arith-emulate-wide-int="function-prefix=emulate_me_" | FileCheck %s
   6
   7 // CHECK-LABEL: func.func @entry()
   8 // CHECK:         {{%.+}} = call @emulate_me_please({{.+}}) : (i64) -> i64
   9 // CHECK-NEXT:    {{%.+}} = call @foo({{.+}}) : (i64) -> i64
  10 func.func @entry() {
  11   %cst0 = arith.constant 0 : i64
  12   func.call @emulate_me_please(%cst0) : (i64) -> (i64)
  13   func.call @foo(%cst0) : (i64) -> (i64)
  14   return
  15 }
  16
  17 // CHECK-LABEL: func.func @emulate_me_please
  18 // 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, i1
  25 // CHECK:         [[RES:%.+]]      = llvm.bitcast {{%.+}} : vector<2xi32> to i64
  26 // CHECK-NEXT:    return [[RES]] : i64
  27 func.func @emulate_me_please(%x : i64) -> i64 {
  28   %r = arith.addi %x, %x : i64
  29   return %r : i64
  30 }
  31
  32 // CHECK-LABEL: func.func @foo
  33 // CHECK-SAME:    ([[ARG:%.+]]: i64) -> i64 {
  34 // CHECK-NEXT:    [[RES:%.+]] = arith.addi [[ARG]], [[ARG]] : i64
  35 // CHECK-NEXT:    return [[RES]] : i64
  36 func.func @foo(%x : i64) -> i64 {
  37   %r = arith.addi %x, %x : i64
  38   return %r : i64
  39 }