llvm/test/CodeGen/LoongArch/select-to-shiftand.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc --mtriple=loongarch32 -mattr=+d < %s | FileCheck %s --check-prefix=LA32
   3 ; RUN: llc --mtriple=loongarch64 -mattr=+d < %s | FileCheck %s --check-prefix=LA64
   4
   5 ;; Compare if positive and select variable or zero.
   6 define i8 @pos_sel_variable_and_zero_i8(i8 signext %a, i8 signext %b) {
   7 ; LA32-LABEL: pos_sel_variable_and_zero_i8:
   8 ; LA32:       # %bb.0:
   9 ; LA32-NEXT:    srai.w $a0, $a0, 7
  10 ; LA32-NEXT:    andn $a0, $a1, $a0
  11 ; LA32-NEXT:    ret
  12 ;
  13 ; LA64-LABEL: pos_sel_variable_and_zero_i8:
  14 ; LA64:       # %bb.0:
  15 ; LA64-NEXT:    srai.d $a0, $a0, 7
  16 ; LA64-NEXT:    andn $a0, $a1, $a0
  17 ; LA64-NEXT:    ret
  18   %cmp = icmp sgt i8 %a, -1
  19   %sel = select i1 %cmp, i8 %b, i8 0
  20   ret i8 %sel
  21 }
  22
  23 define i16 @pos_sel_variable_and_zero_i16(i16 signext %a, i16 signext %b) {
  24 ; LA32-LABEL: pos_sel_variable_and_zero_i16:
  25 ; LA32:       # %bb.0:
  26 ; LA32-NEXT:    srai.w $a0, $a0, 15
  27 ; LA32-NEXT:    andn $a0, $a1, $a0
  28 ; LA32-NEXT:    ret
  29 ;
  30 ; LA64-LABEL: pos_sel_variable_and_zero_i16:
  31 ; LA64:       # %bb.0:
  32 ; LA64-NEXT:    srai.d $a0, $a0, 15
  33 ; LA64-NEXT:    andn $a0, $a1, $a0
  34 ; LA64-NEXT:    ret
  35   %cmp = icmp sgt i16 %a, -1
  36   %sel = select i1 %cmp, i16 %b, i16 0
  37   ret i16 %sel
  38 }
  39
  40 define i32 @pos_sel_variable_and_zero_i32(i32 signext %a, i32 signext %b) {
  41 ; LA32-LABEL: pos_sel_variable_and_zero_i32:
  42 ; LA32:       # %bb.0:
  43 ; LA32-NEXT:    srai.w $a0, $a0, 31
  44 ; LA32-NEXT:    andn $a0, $a1, $a0
  45 ; LA32-NEXT:    ret
  46 ;
  47 ; LA64-LABEL: pos_sel_variable_and_zero_i32:
  48 ; LA64:       # %bb.0:
  49 ; LA64-NEXT:    srai.d $a0, $a0, 31
  50 ; LA64-NEXT:    andn $a0, $a1, $a0
  51 ; LA64-NEXT:    ret
  52   %cmp = icmp sgt i32 %a, -1
  53   %sel = select i1 %cmp, i32 %b, i32 0
  54   ret i32 %sel
  55 }
  56
  57 define i64 @pos_sel_variable_and_zero_i64(i64 signext %a, i64 signext %b) {
  58 ; LA32-LABEL: pos_sel_variable_and_zero_i64:
  59 ; LA32:       # %bb.0:
  60 ; LA32-NEXT:    srai.w $a1, $a1, 31
  61 ; LA32-NEXT:    andn $a0, $a2, $a1
  62 ; LA32-NEXT:    andn $a1, $a3, $a1
  63 ; LA32-NEXT:    ret
  64 ;
  65 ; LA64-LABEL: pos_sel_variable_and_zero_i64:
  66 ; LA64:       # %bb.0:
  67 ; LA64-NEXT:    srai.d $a0, $a0, 63
  68 ; LA64-NEXT:    andn $a0, $a1, $a0
  69 ; LA64-NEXT:    ret
  70   %cmp = icmp sgt i64 %a, -1
  71   %sel = select i1 %cmp, i64 %b, i64 0
  72   ret i64 %sel
  73 }
  74
  75 ;; Compare if not negative or zero and select the same variable as being
  76 ;; compared: smax(a, 0).
  77 define i8 @not_neg_not_zero_sel_same_variable_i8(i8 signext %a) {
  78 ; LA32-LABEL: not_neg_not_zero_sel_same_variable_i8:
  79 ; LA32:       # %bb.0:
  80 ; LA32-NEXT:    srai.w $a1, $a0, 7
  81 ; LA32-NEXT:    andn $a0, $a0, $a1
  82 ; LA32-NEXT:    ret
  83 ;
  84 ; LA64-LABEL: not_neg_not_zero_sel_same_variable_i8:
  85 ; LA64:       # %bb.0:
  86 ; LA64-NEXT:    srai.d $a1, $a0, 7
  87 ; LA64-NEXT:    andn $a0, $a0, $a1
  88 ; LA64-NEXT:    ret
  89   %cmp = icmp sgt i8 %a, 0
  90   %sel = select i1 %cmp, i8 %a, i8 0
  91   ret i8 %sel
  92 }
  93
  94 define i16 @not_neg_not_zero_sel_same_variable_i16(i16 signext %a) {
  95 ; LA32-LABEL: not_neg_not_zero_sel_same_variable_i16:
  96 ; LA32:       # %bb.0:
  97 ; LA32-NEXT:    srai.w $a1, $a0, 15
  98 ; LA32-NEXT:    andn $a0, $a0, $a1
  99 ; LA32-NEXT:    ret
 100 ;
 101 ; LA64-LABEL: not_neg_not_zero_sel_same_variable_i16:
 102 ; LA64:       # %bb.0:
 103 ; LA64-NEXT:    srai.d $a1, $a0, 15
 104 ; LA64-NEXT:    andn $a0, $a0, $a1
 105 ; LA64-NEXT:    ret
 106   %cmp = icmp sgt i16 %a, 0
 107   %sel = select i1 %cmp, i16 %a, i16 0
 108   ret i16 %sel
 109 }
 110
 111 define i32 @not_neg_not_zero_sel_same_variable_i32(i32 signext %a) {
 112 ; LA32-LABEL: not_neg_not_zero_sel_same_variable_i32:
 113 ; LA32:       # %bb.0:
 114 ; LA32-NEXT:    srai.w $a1, $a0, 31
 115 ; LA32-NEXT:    andn $a0, $a0, $a1
 116 ; LA32-NEXT:    ret
 117 ;
 118 ; LA64-LABEL: not_neg_not_zero_sel_same_variable_i32:
 119 ; LA64:       # %bb.0:
 120 ; LA64-NEXT:    srai.d $a1, $a0, 31
 121 ; LA64-NEXT:    andn $a0, $a0, $a1
 122 ; LA64-NEXT:    ret
 123   %cmp = icmp sgt i32 %a, 0
 124   %sel = select i1 %cmp, i32 %a, i32 0
 125   ret i32 %sel
 126 }
 127
 128 define i64 @not_neg_not_zero_sel_same_variable_i64(i64 signext %a) {
 129 ; LA32-LABEL: not_neg_not_zero_sel_same_variable_i64:
 130 ; LA32:       # %bb.0:
 131 ; LA32-NEXT:    srai.w $a2, $a1, 31
 132 ; LA32-NEXT:    andn $a0, $a0, $a2
 133 ; LA32-NEXT:    andn $a1, $a1, $a2
 134 ; LA32-NEXT:    ret
 135 ;
 136 ; LA64-LABEL: not_neg_not_zero_sel_same_variable_i64:
 137 ; LA64:       # %bb.0:
 138 ; LA64-NEXT:    srai.d $a1, $a0, 63
 139 ; LA64-NEXT:    andn $a0, $a0, $a1
 140 ; LA64-NEXT:    ret
 141   %cmp = icmp sgt i64 %a, 0
 142   %sel = select i1 %cmp, i64 %a, i64 0
 143   ret i64 %sel
 144 }
 145
 146 ;; ret = (x-y) > 0 ? x-y : 0
 147 define i8 @sub_clamp_zero_i8(i8 signext %x, i8 signext %y) {
 148 ; LA32-LABEL: sub_clamp_zero_i8:
 149 ; LA32:       # %bb.0:
 150 ; LA32-NEXT:    sub.w $a0, $a0, $a1
 151 ; LA32-NEXT:    ext.w.b $a1, $a0
 152 ; LA32-NEXT:    srai.w $a1, $a1, 7
 153 ; LA32-NEXT:    andn $a0, $a0, $a1
 154 ; LA32-NEXT:    ret
 155 ;
 156 ; LA64-LABEL: sub_clamp_zero_i8:
 157 ; LA64:       # %bb.0:
 158 ; LA64-NEXT:    sub.d $a0, $a0, $a1
 159 ; LA64-NEXT:    ext.w.b $a1, $a0
 160 ; LA64-NEXT:    srai.d $a1, $a1, 7
 161 ; LA64-NEXT:    andn $a0, $a0, $a1
 162 ; LA64-NEXT:    ret
 163   %sub = sub nsw i8 %x, %y
 164   %cmp = icmp sgt i8 %sub, 0
 165   %sel = select i1 %cmp, i8 %sub, i8 0
 166   ret i8 %sel
 167 }
 168
 169 define i16 @sub_clamp_zero_i16(i16 signext %x, i16 signext %y) {
 170 ; LA32-LABEL: sub_clamp_zero_i16:
 171 ; LA32:       # %bb.0:
 172 ; LA32-NEXT:    sub.w $a0, $a0, $a1
 173 ; LA32-NEXT:    ext.w.h $a1, $a0
 174 ; LA32-NEXT:    srai.w $a1, $a1, 15
 175 ; LA32-NEXT:    andn $a0, $a0, $a1
 176 ; LA32-NEXT:    ret
 177 ;
 178 ; LA64-LABEL: sub_clamp_zero_i16:
 179 ; LA64:       # %bb.0:
 180 ; LA64-NEXT:    sub.d $a0, $a0, $a1
 181 ; LA64-NEXT:    ext.w.h $a1, $a0
 182 ; LA64-NEXT:    srai.d $a1, $a1, 15
 183 ; LA64-NEXT:    andn $a0, $a0, $a1
 184 ; LA64-NEXT:    ret
 185   %sub = sub nsw i16 %x, %y
 186   %cmp = icmp sgt i16 %sub, 0
 187   %sel = select i1 %cmp, i16 %sub, i16 0
 188   ret i16 %sel
 189 }
 190
 191 define i32 @sub_clamp_zero_i32(i32 signext %x, i32 signext %y) {
 192 ; LA32-LABEL: sub_clamp_zero_i32:
 193 ; LA32:       # %bb.0:
 194 ; LA32-NEXT:    sub.w $a0, $a0, $a1
 195 ; LA32-NEXT:    srai.w $a1, $a0, 31
 196 ; LA32-NEXT:    andn $a0, $a0, $a1
 197 ; LA32-NEXT:    ret
 198 ;
 199 ; LA64-LABEL: sub_clamp_zero_i32:
 200 ; LA64:       # %bb.0:
 201 ; LA64-NEXT:    sub.w $a0, $a0, $a1
 202 ; LA64-NEXT:    srai.d $a1, $a0, 31
 203 ; LA64-NEXT:    andn $a0, $a0, $a1
 204 ; LA64-NEXT:    ret
 205   %sub = sub nsw i32 %x, %y
 206   %cmp = icmp sgt i32 %sub, 0
 207   %sel = select i1 %cmp, i32 %sub, i32 0
 208   ret i32 %sel
 209 }
 210
 211 define i64 @sub_clamp_zero_i64(i64 signext %x, i64 signext %y) {
 212 ; LA32-LABEL: sub_clamp_zero_i64:
 213 ; LA32:       # %bb.0:
 214 ; LA32-NEXT:    sltu $a4, $a0, $a2
 215 ; LA32-NEXT:    sub.w $a1, $a1, $a3
 216 ; LA32-NEXT:    sub.w $a1, $a1, $a4
 217 ; LA32-NEXT:    sub.w $a0, $a0, $a2
 218 ; LA32-NEXT:    srai.w $a2, $a1, 31
 219 ; LA32-NEXT:    andn $a1, $a1, $a2
 220 ; LA32-NEXT:    andn $a0, $a0, $a2
 221 ; LA32-NEXT:    ret
 222 ;
 223 ; LA64-LABEL: sub_clamp_zero_i64:
 224 ; LA64:       # %bb.0:
 225 ; LA64-NEXT:    sub.d $a0, $a0, $a1
 226 ; LA64-NEXT:    srai.d $a1, $a0, 63
 227 ; LA64-NEXT:    andn $a0, $a0, $a1
 228 ; LA64-NEXT:    ret
 229   %sub = sub nsw i64 %x, %y
 230   %cmp = icmp sgt i64 %sub, 0
 231   %sel = select i1 %cmp, i64 %sub, i64 0
 232   ret i64 %sel
 233 }