llvm/test/CodeGen/AArch64/usub_sat_plus.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple=aarch64-- | FileCheck %s --check-prefixes=CHECK,CHECK-SD
   3 ; RUN: llc < %s -mtriple=aarch64-- -global-isel -global-isel-abort=2 2>&1 | FileCheck %s --check-prefixes=CHECK,CHECK-GI
   4
   5 declare i4 @llvm.usub.sat.i4(i4, i4)
   6 declare i8 @llvm.usub.sat.i8(i8, i8)
   7 declare i16 @llvm.usub.sat.i16(i16, i16)
   8 declare i32 @llvm.usub.sat.i32(i32, i32)
   9 declare i64 @llvm.usub.sat.i64(i64, i64)
  10
  11 define i32 @func32(i32 %x, i32 %y, i32 %z) nounwind {
  12 ; CHECK-SD-LABEL: func32:
  13 ; CHECK-SD:       // %bb.0:
  14 ; CHECK-SD-NEXT:    mul w8, w1, w2
  15 ; CHECK-SD-NEXT:    subs w8, w0, w8
  16 ; CHECK-SD-NEXT:    csel w0, wzr, w8, lo
  17 ; CHECK-SD-NEXT:    ret
  18 ;
  19 ; CHECK-GI-LABEL: func32:
  20 ; CHECK-GI:       // %bb.0:
  21 ; CHECK-GI-NEXT:    mul w8, w1, w2
  22 ; CHECK-GI-NEXT:    subs w8, w0, w8
  23 ; CHECK-GI-NEXT:    cset w9, lo
  24 ; CHECK-GI-NEXT:    tst w9, #0x1
  25 ; CHECK-GI-NEXT:    csel w0, wzr, w8, ne
  26 ; CHECK-GI-NEXT:    ret
  27   %a = mul i32 %y, %z
  28   %tmp = call i32 @llvm.usub.sat.i32(i32 %x, i32 %a)
  29   ret i32 %tmp
  30 }
  31
  32 define i64 @func64(i64 %x, i64 %y, i64 %z) nounwind {
  33 ; CHECK-SD-LABEL: func64:
  34 ; CHECK-SD:       // %bb.0:
  35 ; CHECK-SD-NEXT:    subs x8, x0, x2
  36 ; CHECK-SD-NEXT:    csel x0, xzr, x8, lo
  37 ; CHECK-SD-NEXT:    ret
  38 ;
  39 ; CHECK-GI-LABEL: func64:
  40 ; CHECK-GI:       // %bb.0:
  41 ; CHECK-GI-NEXT:    subs x8, x0, x2
  42 ; CHECK-GI-NEXT:    cset w9, lo
  43 ; CHECK-GI-NEXT:    tst w9, #0x1
  44 ; CHECK-GI-NEXT:    csel x0, xzr, x8, ne
  45 ; CHECK-GI-NEXT:    ret
  46   %a = mul i64 %y, %z
  47   %tmp = call i64 @llvm.usub.sat.i64(i64 %x, i64 %z)
  48   ret i64 %tmp
  49 }
  50
  51 define i16 @func16(i16 %x, i16 %y, i16 %z) nounwind {
  52 ; CHECK-SD-LABEL: func16:
  53 ; CHECK-SD:       // %bb.0:
  54 ; CHECK-SD-NEXT:    mul w8, w1, w2
  55 ; CHECK-SD-NEXT:    and w9, w0, #0xffff
  56 ; CHECK-SD-NEXT:    subs w8, w9, w8, uxth
  57 ; CHECK-SD-NEXT:    csel w0, wzr, w8, lo
  58 ; CHECK-SD-NEXT:    ret
  59 ;
  60 ; CHECK-GI-LABEL: func16:
  61 ; CHECK-GI:       // %bb.0:
  62 ; CHECK-GI-NEXT:    mul w8, w1, w2
  63 ; CHECK-GI-NEXT:    and w9, w0, #0xffff
  64 ; CHECK-GI-NEXT:    sub w8, w9, w8, uxth
  65 ; CHECK-GI-NEXT:    cmp w8, w8, uxth
  66 ; CHECK-GI-NEXT:    csel w0, wzr, w8, ne
  67 ; CHECK-GI-NEXT:    ret
  68   %a = mul i16 %y, %z
  69   %tmp = call i16 @llvm.usub.sat.i16(i16 %x, i16 %a)
  70   ret i16 %tmp
  71 }
  72
  73 define i8 @func8(i8 %x, i8 %y, i8 %z) nounwind {
  74 ; CHECK-SD-LABEL: func8:
  75 ; CHECK-SD:       // %bb.0:
  76 ; CHECK-SD-NEXT:    mul w8, w1, w2
  77 ; CHECK-SD-NEXT:    and w9, w0, #0xff
  78 ; CHECK-SD-NEXT:    subs w8, w9, w8, uxtb
  79 ; CHECK-SD-NEXT:    csel w0, wzr, w8, lo
  80 ; CHECK-SD-NEXT:    ret
  81 ;
  82 ; CHECK-GI-LABEL: func8:
  83 ; CHECK-GI:       // %bb.0:
  84 ; CHECK-GI-NEXT:    mul w8, w1, w2
  85 ; CHECK-GI-NEXT:    and w9, w0, #0xff
  86 ; CHECK-GI-NEXT:    sub w8, w9, w8, uxtb
  87 ; CHECK-GI-NEXT:    cmp w8, w8, uxtb
  88 ; CHECK-GI-NEXT:    csel w0, wzr, w8, ne
  89 ; CHECK-GI-NEXT:    ret
  90   %a = mul i8 %y, %z
  91   %tmp = call i8 @llvm.usub.sat.i8(i8 %x, i8 %a)
  92   ret i8 %tmp
  93 }
  94
  95 define i4 @func4(i4 %x, i4 %y, i4 %z) nounwind {
  96 ; CHECK-SD-LABEL: func4:
  97 ; CHECK-SD:       // %bb.0:
  98 ; CHECK-SD-NEXT:    mul w8, w1, w2
  99 ; CHECK-SD-NEXT:    and w9, w0, #0xf
 100 ; CHECK-SD-NEXT:    and w8, w8, #0xf
 101 ; CHECK-SD-NEXT:    subs w8, w9, w8
 102 ; CHECK-SD-NEXT:    csel w0, wzr, w8, lo
 103 ; CHECK-SD-NEXT:    ret
 104 ;
 105 ; CHECK-GI-LABEL: func4:
 106 ; CHECK-GI:       // %bb.0:
 107 ; CHECK-GI-NEXT:    mul w8, w1, w2
 108 ; CHECK-GI-NEXT:    and w9, w0, #0xf
 109 ; CHECK-GI-NEXT:    and w8, w8, #0xf
 110 ; CHECK-GI-NEXT:    sub w8, w9, w8
 111 ; CHECK-GI-NEXT:    and w9, w8, #0xf
 112 ; CHECK-GI-NEXT:    cmp w8, w9
 113 ; CHECK-GI-NEXT:    csel w0, wzr, w8, ne
 114 ; CHECK-GI-NEXT:    ret
 115   %a = mul i4 %y, %z
 116   %tmp = call i4 @llvm.usub.sat.i4(i4 %x, i4 %a)
 117   ret i4 %tmp
 118 }
 119 ;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:
 120 ; CHECK: {{.*}}