test/CodeGen/ARM64/shifted-sext.ll

   1 ; RUN: llc -march=arm64 -mtriple=arm64-apple-ios < %s | FileCheck %s
   2 ;
   3 ; <rdar://problem/13820218>
   4
   5 define signext i16 @extendedLeftShiftcharToshortBy4(i8 signext %a) nounwind readnone ssp {
   6 entry:
   7 ; CHECK-LABEL: extendedLeftShiftcharToshortBy4:
   8 ; CHECK: add [[REG:w[0-9]+]], w0, #1
   9 ; CHECK: sbfm w0, [[REG]], #28, #7
  10   %inc = add i8 %a, 1
  11   %conv1 = sext i8 %inc to i32
  12   %shl = shl nsw i32 %conv1, 4
  13   %conv2 = trunc i32 %shl to i16
  14   ret i16 %conv2
  15 }
  16
  17 define signext i16 @extendedRightShiftcharToshortBy4(i8 signext %a) nounwind readnone ssp {
  18 entry:
  19 ; CHECK-LABEL: extendedRightShiftcharToshortBy4:
  20 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  21 ; CHECK: sbfm w0, [[REG]], #4, #7
  22   %inc = add i8 %a, 1
  23   %conv1 = sext i8 %inc to i32
  24   %shr4 = lshr i32 %conv1, 4
  25   %conv2 = trunc i32 %shr4 to i16
  26   ret i16 %conv2
  27 }
  28
  29 define signext i16 @extendedLeftShiftcharToshortBy8(i8 signext %a) nounwind readnone ssp {
  30 entry:
  31 ; CHECK-LABEL: extendedLeftShiftcharToshortBy8:
  32 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  33 ; CHECK: sbfm w0, [[REG]], #24, #7
  34   %inc = add i8 %a, 1
  35   %conv1 = sext i8 %inc to i32
  36   %shl = shl nsw i32 %conv1, 8
  37   %conv2 = trunc i32 %shl to i16
  38   ret i16 %conv2
  39 }
  40
  41 define signext i16 @extendedRightShiftcharToshortBy8(i8 signext %a) nounwind readnone ssp {
  42 entry:
  43 ; CHECK-LABEL: extendedRightShiftcharToshortBy8:
  44 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  45 ; CHECK: sxtb [[REG]], [[REG]]
  46 ; CHECK: asr w0, [[REG]], #8
  47   %inc = add i8 %a, 1
  48   %conv1 = sext i8 %inc to i32
  49   %shr4 = lshr i32 %conv1, 8
  50   %conv2 = trunc i32 %shr4 to i16
  51   ret i16 %conv2
  52 }
  53
  54 define i32 @extendedLeftShiftcharTointBy4(i8 signext %a) nounwind readnone ssp {
  55 entry:
  56 ; CHECK-LABEL: extendedLeftShiftcharTointBy4:
  57 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  58 ; CHECK: sbfm w0, [[REG]], #28, #7
  59   %inc = add i8 %a, 1
  60   %conv = sext i8 %inc to i32
  61   %shl = shl nsw i32 %conv, 4
  62   ret i32 %shl
  63 }
  64
  65 define i32 @extendedRightShiftcharTointBy4(i8 signext %a) nounwind readnone ssp {
  66 entry:
  67 ; CHECK-LABEL: extendedRightShiftcharTointBy4:
  68 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  69 ; CHECK: sbfm w0, [[REG]], #4, #7
  70   %inc = add i8 %a, 1
  71   %conv = sext i8 %inc to i32
  72   %shr = ashr i32 %conv, 4
  73   ret i32 %shr
  74 }
  75
  76 define i32 @extendedLeftShiftcharTointBy8(i8 signext %a) nounwind readnone ssp {
  77 entry:
  78 ; CHECK-LABEL: extendedLeftShiftcharTointBy8:
  79 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  80 ; CHECK: sbfm w0, [[REG]], #24, #7
  81   %inc = add i8 %a, 1
  82   %conv = sext i8 %inc to i32
  83   %shl = shl nsw i32 %conv, 8
  84   ret i32 %shl
  85 }
  86
  87 define i32 @extendedRightShiftcharTointBy8(i8 signext %a) nounwind readnone ssp {
  88 entry:
  89 ; CHECK-LABEL: extendedRightShiftcharTointBy8:
  90 ; CHECK: add [[REG:w[0-9]+]], w0, #1
  91 ; CHECK: sxtb [[REG]], [[REG]]
  92 ; CHECK: asr w0, [[REG]], #8
  93   %inc = add i8 %a, 1
  94   %conv = sext i8 %inc to i32
  95   %shr = ashr i32 %conv, 8
  96   ret i32 %shr
  97 }
  98
  99 define i64 @extendedLeftShiftcharToint64By4(i8 signext %a) nounwind readnone ssp {
 100 entry:
 101 ; CHECK-LABEL: extendedLeftShiftcharToint64By4:
 102 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 103 ; CHECK: sbfm x0, x[[REG]], #60, #7
 104   %inc = add i8 %a, 1
 105   %conv = sext i8 %inc to i64
 106   %shl = shl nsw i64 %conv, 4
 107   ret i64 %shl
 108 }
 109
 110 define i64 @extendedRightShiftcharToint64By4(i8 signext %a) nounwind readnone ssp {
 111 entry:
 112 ; CHECK-LABEL: extendedRightShiftcharToint64By4:
 113 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 114 ; CHECK: sbfm x0, x[[REG]], #4, #7
 115   %inc = add i8 %a, 1
 116   %conv = sext i8 %inc to i64
 117   %shr = ashr i64 %conv, 4
 118   ret i64 %shr
 119 }
 120
 121 define i64 @extendedLeftShiftcharToint64By8(i8 signext %a) nounwind readnone ssp {
 122 entry:
 123 ; CHECK-LABEL: extendedLeftShiftcharToint64By8:
 124 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 125 ; CHECK: sbfm x0, x[[REG]], #56, #7
 126   %inc = add i8 %a, 1
 127   %conv = sext i8 %inc to i64
 128   %shl = shl nsw i64 %conv, 8
 129   ret i64 %shl
 130 }
 131
 132 define i64 @extendedRightShiftcharToint64By8(i8 signext %a) nounwind readnone ssp {
 133 entry:
 134 ; CHECK-LABEL: extendedRightShiftcharToint64By8:
 135 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 136 ; CHECK: sxtb x[[REG]], x[[REG]]
 137 ; CHECK: asr x0, x[[REG]], #8
 138   %inc = add i8 %a, 1
 139   %conv = sext i8 %inc to i64
 140   %shr = ashr i64 %conv, 8
 141   ret i64 %shr
 142 }
 143
 144 define i32 @extendedLeftShiftshortTointBy4(i16 signext %a) nounwind readnone ssp {
 145 entry:
 146 ; CHECK-LABEL: extendedLeftShiftshortTointBy4:
 147 ; CHECK: add [[REG:w[0-9]+]], w0, #1
 148 ; CHECK: sbfm w0, [[REG]], #28, #15
 149   %inc = add i16 %a, 1
 150   %conv = sext i16 %inc to i32
 151   %shl = shl nsw i32 %conv, 4
 152   ret i32 %shl
 153 }
 154
 155 define i32 @extendedRightShiftshortTointBy4(i16 signext %a) nounwind readnone ssp {
 156 entry:
 157 ; CHECK-LABEL: extendedRightShiftshortTointBy4:
 158 ; CHECK: add [[REG:w[0-9]+]], w0, #1
 159 ; CHECK: sbfm w0, [[REG]], #4, #15
 160   %inc = add i16 %a, 1
 161   %conv = sext i16 %inc to i32
 162   %shr = ashr i32 %conv, 4
 163   ret i32 %shr
 164 }
 165
 166 define i32 @extendedLeftShiftshortTointBy16(i16 signext %a) nounwind readnone ssp {
 167 entry:
 168 ; CHECK-LABEL: extendedLeftShiftshortTointBy16:
 169 ; CHECK: add [[REG:w[0-9]+]], w0, #1
 170 ; CHECK: lsl w0, [[REG]], #16
 171   %inc = add i16 %a, 1
 172   %conv2 = zext i16 %inc to i32
 173   %shl = shl nuw i32 %conv2, 16
 174   ret i32 %shl
 175 }
 176
 177 define i32 @extendedRightShiftshortTointBy16(i16 signext %a) nounwind readnone ssp {
 178 entry:
 179 ; CHECK-LABEL: extendedRightShiftshortTointBy16:
 180 ; CHECK: add [[REG:w[0-9]+]], w0, #1
 181 ; CHECK: sxth [[REG]], [[REG]]
 182 ; CHECK: asr w0, [[REG]], #16
 183   %inc = add i16 %a, 1
 184   %conv = sext i16 %inc to i32
 185   %shr = ashr i32 %conv, 16
 186   ret i32 %shr
 187 }
 188
 189 define i64 @extendedLeftShiftshortToint64By4(i16 signext %a) nounwind readnone ssp {
 190 entry:
 191 ; CHECK-LABEL: extendedLeftShiftshortToint64By4:
 192 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 193 ; CHECK: sbfm x0, x[[REG]], #60, #15
 194   %inc = add i16 %a, 1
 195   %conv = sext i16 %inc to i64
 196   %shl = shl nsw i64 %conv, 4
 197   ret i64 %shl
 198 }
 199
 200 define i64 @extendedRightShiftshortToint64By4(i16 signext %a) nounwind readnone ssp {
 201 entry:
 202 ; CHECK-LABEL: extendedRightShiftshortToint64By4:
 203 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 204 ; CHECK: sbfm x0, x[[REG]], #4, #15
 205   %inc = add i16 %a, 1
 206   %conv = sext i16 %inc to i64
 207   %shr = ashr i64 %conv, 4
 208   ret i64 %shr
 209 }
 210
 211 define i64 @extendedLeftShiftshortToint64By16(i16 signext %a) nounwind readnone ssp {
 212 entry:
 213 ; CHECK-LABEL: extendedLeftShiftshortToint64By16:
 214 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 215 ; CHECK: sbfm x0, x[[REG]], #48, #15
 216   %inc = add i16 %a, 1
 217   %conv = sext i16 %inc to i64
 218   %shl = shl nsw i64 %conv, 16
 219   ret i64 %shl
 220 }
 221
 222 define i64 @extendedRightShiftshortToint64By16(i16 signext %a) nounwind readnone ssp {
 223 entry:
 224 ; CHECK-LABEL: extendedRightShiftshortToint64By16:
 225 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 226 ; CHECK: sxth x[[REG]], x[[REG]]
 227 ; CHECK: asr x0, x[[REG]], #16
 228   %inc = add i16 %a, 1
 229   %conv = sext i16 %inc to i64
 230   %shr = ashr i64 %conv, 16
 231   ret i64 %shr
 232 }
 233
 234 define i64 @extendedLeftShiftintToint64By4(i32 %a) nounwind readnone ssp {
 235 entry:
 236 ; CHECK-LABEL: extendedLeftShiftintToint64By4:
 237 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 238 ; CHECK: sbfm x0, x[[REG]], #60, #31
 239   %inc = add nsw i32 %a, 1
 240   %conv = sext i32 %inc to i64
 241   %shl = shl nsw i64 %conv, 4
 242   ret i64 %shl
 243 }
 244
 245 define i64 @extendedRightShiftintToint64By4(i32 %a) nounwind readnone ssp {
 246 entry:
 247 ; CHECK-LABEL: extendedRightShiftintToint64By4:
 248 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 249 ; CHECK: sbfm x0, x[[REG]], #4, #31
 250   %inc = add nsw i32 %a, 1
 251   %conv = sext i32 %inc to i64
 252   %shr = ashr i64 %conv, 4
 253   ret i64 %shr
 254 }
 255
 256 define i64 @extendedLeftShiftintToint64By32(i32 %a) nounwind readnone ssp {
 257 entry:
 258 ; CHECK-LABEL: extendedLeftShiftintToint64By32:
 259 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 260 ; CHECK: lsl x0, x[[REG]], #32
 261   %inc = add nsw i32 %a, 1
 262   %conv2 = zext i32 %inc to i64
 263   %shl = shl nuw i64 %conv2, 32
 264   ret i64 %shl
 265 }
 266
 267 define i64 @extendedRightShiftintToint64By32(i32 %a) nounwind readnone ssp {
 268 entry:
 269 ; CHECK-LABEL: extendedRightShiftintToint64By32:
 270 ; CHECK: add w[[REG:[0-9]+]], w0, #1
 271 ; CHECK: sxtw x[[REG]], x[[REG]]
 272 ; CHECK: asr x0, x[[REG]], #32
 273   %inc = add nsw i32 %a, 1
 274   %conv = sext i32 %inc to i64
 275   %shr = ashr i64 %conv, 32
 276   ret i64 %shr
 277 }