llvm/test/CodeGen/AArch64/uadd_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.uadd.sat.i4(i4, i4)
   6 declare i8 @llvm.uadd.sat.i8(i8, i8)
   7 declare i16 @llvm.uadd.sat.i16(i16, i16)
   8 declare i32 @llvm.uadd.sat.i32(i32, i32)
   9 declare i64 @llvm.uadd.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:    adds w8, w0, w8
  16 ; CHECK-SD-NEXT:    csinv w0, w8, wzr, 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:    adds w8, w0, w8
  23 ; CHECK-GI-NEXT:    cset w9, hs
  24 ; CHECK-GI-NEXT:    tst w9, #0x1
  25 ; CHECK-GI-NEXT:    csinv w0, w8, wzr, eq
  26 ; CHECK-GI-NEXT:    ret
  27   %a = mul i32 %y, %z
  28   %tmp = call i32 @llvm.uadd.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:    adds x8, x0, x2
  36 ; CHECK-SD-NEXT:    csinv x0, x8, xzr, lo
  37 ; CHECK-SD-NEXT:    ret
  38 ;
  39 ; CHECK-GI-LABEL: func64:
  40 ; CHECK-GI:       // %bb.0:
  41 ; CHECK-GI-NEXT:    adds x8, x0, x2
  42 ; CHECK-GI-NEXT:    cset w9, hs
  43 ; CHECK-GI-NEXT:    tst w9, #0x1
  44 ; CHECK-GI-NEXT:    csinv x0, x8, xzr, eq
  45 ; CHECK-GI-NEXT:    ret
  46   %a = mul i64 %y, %z
  47   %tmp = call i64 @llvm.uadd.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:    add w8, w9, w8, uxth
  57 ; CHECK-SD-NEXT:    mov w9, #65535 // =0xffff
  58 ; CHECK-SD-NEXT:    cmp w8, w9
  59 ; CHECK-SD-NEXT:    csel w0, w8, w9, lo
  60 ; CHECK-SD-NEXT:    ret
  61 ;
  62 ; CHECK-GI-LABEL: func16:
  63 ; CHECK-GI:       // %bb.0:
  64 ; CHECK-GI-NEXT:    mul w8, w1, w2
  65 ; CHECK-GI-NEXT:    and w8, w8, #0xffff
  66 ; CHECK-GI-NEXT:    add w8, w8, w0, uxth
  67 ; CHECK-GI-NEXT:    cmp w8, w8, uxth
  68 ; CHECK-GI-NEXT:    csinv w0, w8, wzr, eq
  69 ; CHECK-GI-NEXT:    ret
  70   %a = mul i16 %y, %z
  71   %tmp = call i16 @llvm.uadd.sat.i16(i16 %x, i16 %a)
  72   ret i16 %tmp
  73 }
  74
  75 define i8 @func8(i8 %x, i8 %y, i8 %z) nounwind {
  76 ; CHECK-SD-LABEL: func8:
  77 ; CHECK-SD:       // %bb.0:
  78 ; CHECK-SD-NEXT:    mul w8, w1, w2
  79 ; CHECK-SD-NEXT:    and w9, w0, #0xff
  80 ; CHECK-SD-NEXT:    add w8, w9, w8, uxtb
  81 ; CHECK-SD-NEXT:    mov w9, #255 // =0xff
  82 ; CHECK-SD-NEXT:    cmp w8, #255
  83 ; CHECK-SD-NEXT:    csel w0, w8, w9, lo
  84 ; CHECK-SD-NEXT:    ret
  85 ;
  86 ; CHECK-GI-LABEL: func8:
  87 ; CHECK-GI:       // %bb.0:
  88 ; CHECK-GI-NEXT:    mul w8, w1, w2
  89 ; CHECK-GI-NEXT:    and w8, w8, #0xff
  90 ; CHECK-GI-NEXT:    add w8, w8, w0, uxtb
  91 ; CHECK-GI-NEXT:    cmp w8, w8, uxtb
  92 ; CHECK-GI-NEXT:    csinv w0, w8, wzr, eq
  93 ; CHECK-GI-NEXT:    ret
  94   %a = mul i8 %y, %z
  95   %tmp = call i8 @llvm.uadd.sat.i8(i8 %x, i8 %a)
  96   ret i8 %tmp
  97 }
  98
  99 define i4 @func4(i4 %x, i4 %y, i4 %z) nounwind {
 100 ; CHECK-SD-LABEL: func4:
 101 ; CHECK-SD:       // %bb.0:
 102 ; CHECK-SD-NEXT:    mul w8, w1, w2
 103 ; CHECK-SD-NEXT:    and w9, w0, #0xf
 104 ; CHECK-SD-NEXT:    and w8, w8, #0xf
 105 ; CHECK-SD-NEXT:    add w8, w9, w8
 106 ; CHECK-SD-NEXT:    mov w9, #15 // =0xf
 107 ; CHECK-SD-NEXT:    cmp w8, #15
 108 ; CHECK-SD-NEXT:    csel w0, w8, w9, lo
 109 ; CHECK-SD-NEXT:    ret
 110 ;
 111 ; CHECK-GI-LABEL: func4:
 112 ; CHECK-GI:       // %bb.0:
 113 ; CHECK-GI-NEXT:    mul w8, w1, w2
 114 ; CHECK-GI-NEXT:    and w9, w0, #0xf
 115 ; CHECK-GI-NEXT:    mov w10, #15 // =0xf
 116 ; CHECK-GI-NEXT:    and w8, w8, #0xf
 117 ; CHECK-GI-NEXT:    add w8, w9, w8
 118 ; CHECK-GI-NEXT:    and w9, w8, #0xf
 119 ; CHECK-GI-NEXT:    cmp w8, w9
 120 ; CHECK-GI-NEXT:    csel w0, w10, w8, ne
 121 ; CHECK-GI-NEXT:    ret
 122   %a = mul i4 %y, %z
 123   %tmp = call i4 @llvm.uadd.sat.i4(i4 %x, i4 %a)
 124   ret i4 %tmp
 125 }
 126 ;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:
 127 ; CHECK: {{.*}}