test/Transforms/InstCombine/shift.ll

   1 ; This test makes sure that these instructions are properly eliminated.
   2 ;
   3 ; RUN: opt < %s -instcombine -S | FileCheck %s
   4
   5 define i32 @test1(i32 %A) {
   6 ; CHECK: @test1
   7 ; CHECK: ret i32 %A
   8         %B = shl i32 %A, 0              ; <i32> [#uses=1]
   9         ret i32 %B
  10 }
  11
  12 define i32 @test2(i8 %A) {
  13 ; CHECK: @test2
  14 ; CHECK: ret i32 0
  15         %shift.upgrd.1 = zext i8 %A to i32              ; <i32> [#uses=1]
  16         %B = shl i32 0, %shift.upgrd.1          ; <i32> [#uses=1]
  17         ret i32 %B
  18 }
  19
  20 define i32 @test3(i32 %A) {
  21 ; CHECK: @test3
  22 ; CHECK: ret i32 %A
  23         %B = ashr i32 %A, 0             ; <i32> [#uses=1]
  24         ret i32 %B
  25 }
  26
  27 define i32 @test4(i8 %A) {
  28 ; CHECK: @test4
  29 ; CHECK: ret i32 0
  30         %shift.upgrd.2 = zext i8 %A to i32              ; <i32> [#uses=1]
  31         %B = ashr i32 0, %shift.upgrd.2         ; <i32> [#uses=1]
  32         ret i32 %B
  33 }
  34
  35
  36 define i32 @test5(i32 %A) {
  37 ; CHECK: @test5
  38 ; CHECK: ret i32 0
  39         %B = lshr i32 %A, 32  ;; shift all bits out
  40         ret i32 %B
  41 }
  42
  43 define i32 @test5a(i32 %A) {
  44 ; CHECK: @test5a
  45 ; CHECK: ret i32 0
  46         %B = shl i32 %A, 32     ;; shift all bits out
  47         ret i32 %B
  48 }
  49
  50 define i32 @test6(i32 %A) {
  51 ; CHECK: @test6
  52 ; CHECK-NEXT: mul i32 %A, 6
  53 ; CHECK-NEXT: ret i32
  54         %B = shl i32 %A, 1      ;; convert to an mul instruction
  55         %C = mul i32 %B, 3
  56         ret i32 %C
  57 }
  58
  59 define i32 @test7(i8 %A) {
  60 ; CHECK: @test7
  61 ; CHECK-NEXT: ret i32 -1
  62         %shift.upgrd.3 = zext i8 %A to i32
  63         %B = ashr i32 -1, %shift.upgrd.3  ;; Always equal to -1
  64         ret i32 %B
  65 }
  66
  67 ;; (A << 5) << 3 === A << 8 == 0
  68 define i8 @test8(i8 %A) {
  69 ; CHECK: @test8
  70 ; CHECK: ret i8 0
  71         %B = shl i8 %A, 5               ; <i8> [#uses=1]
  72         %C = shl i8 %B, 3               ; <i8> [#uses=1]
  73         ret i8 %C
  74 }
  75
  76 ;; (A << 7) >> 7 === A & 1
  77 define i8 @test9(i8 %A) {
  78 ; CHECK: @test9
  79 ; CHECK-NEXT: and i8 %A, 1
  80 ; CHECK-NEXT: ret i8
  81         %B = shl i8 %A, 7               ; <i8> [#uses=1]
  82         %C = lshr i8 %B, 7              ; <i8> [#uses=1]
  83         ret i8 %C
  84 }
  85
  86 ;; (A >> 7) << 7 === A & 128
  87 define i8 @test10(i8 %A) {
  88 ; CHECK: @test10
  89 ; CHECK-NEXT: and i8 %A, -128
  90 ; CHECK-NEXT: ret i8
  91         %B = lshr i8 %A, 7              ; <i8> [#uses=1]
  92         %C = shl i8 %B, 7               ; <i8> [#uses=1]
  93         ret i8 %C
  94 }
  95
  96 ;; (A >> 3) << 4 === (A & 0x1F) << 1
  97 define i8 @test11(i8 %A) {
  98 ; CHECK: @test11
  99 ; CHECK-NEXT: mul i8 %A, 6
 100 ; CHECK-NEXT: and i8
 101 ; CHECK-NEXT: ret i8
 102         %a = mul i8 %A, 3               ; <i8> [#uses=1]
 103         %B = lshr i8 %a, 3              ; <i8> [#uses=1]
 104         %C = shl i8 %B, 4               ; <i8> [#uses=1]
 105         ret i8 %C
 106 }
 107
 108 ;; (A >> 8) << 8 === A & -256
 109 define i32 @test12(i32 %A) {
 110 ; CHECK: @test12
 111 ; CHECK-NEXT: and i32 %A, -256
 112 ; CHECK-NEXT: ret i32
 113         %B = ashr i32 %A, 8             ; <i32> [#uses=1]
 114         %C = shl i32 %B, 8              ; <i32> [#uses=1]
 115         ret i32 %C
 116 }
 117
 118 ;; (A >> 3) << 4 === (A & -8) * 2
 119 define i8 @test13(i8 %A) {
 120 ; CHECK: @test13
 121 ; CHECK-NEXT: mul i8 %A, 6
 122 ; CHECK-NEXT: and i8
 123 ; CHECK-NEXT: ret i8
 124         %a = mul i8 %A, 3               ; <i8> [#uses=1]
 125         %B = ashr i8 %a, 3              ; <i8> [#uses=1]
 126         %C = shl i8 %B, 4               ; <i8> [#uses=1]
 127         ret i8 %C
 128 }
 129
 130 ;; D = ((B | 1234) << 4) === ((B << 4)|(1234 << 4)
 131 define i32 @test14(i32 %A) {
 132 ; CHECK: @test14
 133 ; CHECK-NEXT: or i32 %A, 19744
 134 ; CHECK-NEXT: and i32
 135 ; CHECK-NEXT: ret i32
 136         %B = lshr i32 %A, 4             ; <i32> [#uses=1]
 137         %C = or i32 %B, 1234            ; <i32> [#uses=1]
 138         %D = shl i32 %C, 4              ; <i32> [#uses=1]
 139         ret i32 %D
 140 }
 141
 142 ;; D = ((B | 1234) << 4) === ((B << 4)|(1234 << 4)
 143 define i32 @test14a(i32 %A) {
 144 ; CHECK: @test14a
 145 ; CHECK-NEXT: and i32 %A, 77
 146 ; CHECK-NEXT: ret i32
 147         %B = shl i32 %A, 4              ; <i32> [#uses=1]
 148         %C = and i32 %B, 1234           ; <i32> [#uses=1]
 149         %D = lshr i32 %C, 4             ; <i32> [#uses=1]
 150         ret i32 %D
 151 }
 152
 153 define i32 @test15(i1 %C) {
 154 ; CHECK: @test15
 155 ; CHECK-NEXT: select i1 %C, i32 12, i32 4
 156 ; CHECK-NEXT: ret i32
 157         %A = select i1 %C, i32 3, i32 1         ; <i32> [#uses=1]
 158         %V = shl i32 %A, 2              ; <i32> [#uses=1]
 159         ret i32 %V
 160 }
 161
 162 define i32 @test15a(i1 %C) {
 163 ; CHECK: @test15a
 164 ; CHECK-NEXT: select i1 %C, i32 512, i32 128
 165 ; CHECK-NEXT: ret i32
 166         %A = select i1 %C, i8 3, i8 1           ; <i8> [#uses=1]
 167         %shift.upgrd.4 = zext i8 %A to i32              ; <i32> [#uses=1]
 168         %V = shl i32 64, %shift.upgrd.4         ; <i32> [#uses=1]
 169         ret i32 %V
 170 }
 171
 172 define i1 @test16(i32 %X) {
 173 ; CHECK: @test16
 174 ; CHECK-NEXT: and i32 %X, 16
 175 ; CHECK-NEXT: icmp ne i32
 176 ; CHECK-NEXT: ret i1
 177         %tmp.3 = ashr i32 %X, 4
 178         %tmp.6 = and i32 %tmp.3, 1
 179         %tmp.7 = icmp ne i32 %tmp.6, 0
 180         ret i1 %tmp.7
 181 }
 182
 183 define i1 @test17(i32 %A) {
 184 ; CHECK: @test17
 185 ; CHECK-NEXT: and i32 %A, -8
 186 ; CHECK-NEXT: icmp eq i32
 187 ; CHECK-NEXT: ret i1
 188         %B = lshr i32 %A, 3             ; <i32> [#uses=1]
 189         %C = icmp eq i32 %B, 1234               ; <i1> [#uses=1]
 190         ret i1 %C
 191 }
 192
 193
 194 define i1 @test18(i8 %A) {
 195 ; CHECK: @test18
 196 ; CHECK: ret i1 false
 197
 198         %B = lshr i8 %A, 7              ; <i8> [#uses=1]
 199         ;; false
 200         %C = icmp eq i8 %B, 123         ; <i1> [#uses=1]
 201         ret i1 %C
 202 }
 203
 204 define i1 @test19(i32 %A) {
 205 ; CHECK: @test19
 206 ; CHECK-NEXT: icmp ult i32 %A, 4
 207 ; CHECK-NEXT: ret i1
 208         %B = ashr i32 %A, 2             ; <i32> [#uses=1]
 209         ;; (X & -4) == 0
 210         %C = icmp eq i32 %B, 0          ; <i1> [#uses=1]
 211         ret i1 %C
 212 }
 213
 214
 215 define i1 @test19a(i32 %A) {
 216 ; CHECK: @test19a
 217 ; CHECK-NEXT: and i32 %A, -4
 218 ; CHECK-NEXT: icmp eq i32
 219 ; CHECK-NEXT: ret i1
 220         %B = ashr i32 %A, 2             ; <i32> [#uses=1]
 221         ;; (X & -4) == -4
 222         %C = icmp eq i32 %B, -1         ; <i1> [#uses=1]
 223         ret i1 %C
 224 }
 225
 226 define i1 @test20(i8 %A) {
 227 ; CHECK: @test20
 228 ; CHECK: ret i1 false
 229         %B = ashr i8 %A, 7              ; <i8> [#uses=1]
 230         ;; false
 231         %C = icmp eq i8 %B, 123         ; <i1> [#uses=1]
 232         ret i1 %C
 233 }
 234
 235 define i1 @test21(i8 %A) {
 236 ; CHECK: @test21
 237 ; CHECK-NEXT: and i8 %A, 15
 238 ; CHECK-NEXT: icmp eq i8
 239 ; CHECK-NEXT: ret i1
 240         %B = shl i8 %A, 4               ; <i8> [#uses=1]
 241         %C = icmp eq i8 %B, -128                ; <i1> [#uses=1]
 242         ret i1 %C
 243 }
 244
 245 define i1 @test22(i8 %A) {
 246 ; CHECK: @test22
 247 ; CHECK-NEXT: and i8 %A, 15
 248 ; CHECK-NEXT: icmp eq i8
 249 ; CHECK-NEXT: ret i1
 250         %B = shl i8 %A, 4               ; <i8> [#uses=1]
 251         %C = icmp eq i8 %B, 0           ; <i1> [#uses=1]
 252         ret i1 %C
 253 }
 254
 255 define i8 @test23(i32 %A) {
 256 ; CHECK: @test23
 257 ; CHECK-NEXT: trunc i32 %A to i8
 258 ; CHECK-NEXT: ret i8
 259
 260         ;; casts not needed
 261         %B = shl i32 %A, 24             ; <i32> [#uses=1]
 262         %C = ashr i32 %B, 24            ; <i32> [#uses=1]
 263         %D = trunc i32 %C to i8         ; <i8> [#uses=1]
 264         ret i8 %D
 265 }
 266
 267 define i8 @test24(i8 %X) {
 268 ; CHECK: @test24
 269 ; CHECK-NEXT: and i8 %X, 3
 270 ; CHECK-NEXT: ret i8
 271         %Y = and i8 %X, -5              ; <i8> [#uses=1]
 272         %Z = shl i8 %Y, 5               ; <i8> [#uses=1]
 273         %Q = ashr i8 %Z, 5              ; <i8> [#uses=1]
 274         ret i8 %Q
 275 }
 276
 277 define i32 @test25(i32 %tmp.2, i32 %AA) {
 278 ; CHECK: @test25
 279 ; CHECK-NEXT: and i32 %tmp.2, -131072
 280 ; CHECK-NEXT: add i32 %{{[^,]*}}, %AA
 281 ; CHECK-NEXT: and i32 %{{[^,]*}}, -131072
 282 ; CHECK-NEXT: ret i32
 283         %x = lshr i32 %AA, 17           ; <i32> [#uses=1]
 284         %tmp.3 = lshr i32 %tmp.2, 17            ; <i32> [#uses=1]
 285         %tmp.5 = add i32 %tmp.3, %x             ; <i32> [#uses=1]
 286         %tmp.6 = shl i32 %tmp.5, 17             ; <i32> [#uses=1]
 287         ret i32 %tmp.6
 288 }
 289
 290 ;; handle casts between shifts.
 291 define i32 @test26(i32 %A) {
 292 ; CHECK: @test26
 293 ; CHECK-NEXT: and i32 %A, -2
 294 ; CHECK-NEXT: ret i32
 295         %B = lshr i32 %A, 1             ; <i32> [#uses=1]
 296         %C = bitcast i32 %B to i32              ; <i32> [#uses=1]
 297         %D = shl i32 %C, 1              ; <i32> [#uses=1]
 298         ret i32 %D
 299 }
 300
 301
 302 define i1 @test27(i32 %x) nounwind {
 303 ; CHECK: @test27
 304 ; CHECK-NEXT: and i32 %x, 8
 305 ; CHECK-NEXT: icmp ne i32
 306 ; CHECK-NEXT: ret i1
 307   %y = lshr i32 %x, 3
 308   %z = trunc i32 %y to i1
 309   ret i1 %z
 310 }
 311
 312 define i8 @test28(i8 %x) {
 313 entry:
 314 ; CHECK: @test28
 315 ; CHECK:     icmp slt i8 %x, 0
 316 ; CHECK-NEXT:     br i1
 317         %tmp1 = lshr i8 %x, 7
 318         %cond1 = icmp ne i8 %tmp1, 0
 319         br i1 %cond1, label %bb1, label %bb2
 320
 321 bb1:
 322         ret i8 0
 323
 324 bb2:
 325         ret i8 1
 326 }
 327
 328 define i8 @test28a(i8 %x, i8 %y) {
 329 entry:
 330 ; This shouldn't be transformed.
 331 ; CHECK: @test28a
 332 ; CHECK:     %tmp1 = lshr i8 %x, 7
 333 ; CHECK:     %cond1 = icmp eq i8 %tmp1, 0
 334 ; CHECK:     br i1 %cond1, label %bb2, label %bb1
 335         %tmp1 = lshr i8 %x, 7
 336         %cond1 = icmp ne i8 %tmp1, 0
 337         br i1 %cond1, label %bb1, label %bb2
 338 bb1:
 339         ret i8 %tmp1
 340 bb2:
 341         %tmp2 = add i8 %tmp1, %y
 342         ret i8 %tmp2
 343 }
 344
 345