test/Transforms/InstCombine/add2.ll

   1 ; RUN: opt < %s -instcombine -S | FileCheck %s
   2
   3 define i64 @test1(i64 %A, i32 %B) {
   4         %tmp12 = zext i32 %B to i64
   5         %tmp3 = shl i64 %tmp12, 32
   6         %tmp5 = add i64 %tmp3, %A
   7         %tmp6 = and i64 %tmp5, 123
   8         ret i64 %tmp6
   9 ; CHECK-LABEL: @test1(
  10 ; CHECK-NEXT: and i64 %A, 123
  11 ; CHECK-NEXT: ret i64
  12 }
  13
  14 define i32 @test2(i32 %A) {
  15   %B = and i32 %A, 7
  16   %C = and i32 %A, 32
  17   %F = add i32 %B, %C
  18   ret i32 %F
  19 ; CHECK-LABEL: @test2(
  20 ; CHECK-NEXT: and i32 %A, 39
  21 ; CHECK-NEXT: ret i32
  22 }
  23
  24 define i32 @test3(i32 %A) {
  25   %B = and i32 %A, 128
  26   %C = lshr i32 %A, 30
  27   %F = add i32 %B, %C
  28   ret i32 %F
  29 ; CHECK-LABEL: @test3(
  30 ; CHECK-NEXT: and
  31 ; CHECK-NEXT: lshr
  32 ; CHECK-NEXT: or i32 %B, %C
  33 ; CHECK-NEXT: ret i32
  34 }
  35
  36 define i32 @test4(i32 %A) {
  37   %B = add nuw i32 %A, %A
  38   ret i32 %B
  39 ; CHECK-LABEL: @test4(
  40 ; CHECK-NEXT: %B = shl nuw i32 %A, 1
  41 ; CHECK-NEXT: ret i32 %B
  42 }
  43
  44 define <2 x i1> @test5(<2 x i1> %A, <2 x i1> %B) {
  45   %add = add <2 x i1> %A, %B
  46   ret <2 x i1> %add
  47 ; CHECK-LABEL: @test5(
  48 ; CHECK-NEXT: %add = xor <2 x i1> %A, %B
  49 ; CHECK-NEXT: ret <2 x i1> %add
  50 }
  51
  52 define <2 x i64> @test6(<2 x i64> %A) {
  53   %shl = shl <2 x i64> %A, <i64 2, i64 3>
  54   %add = add <2 x i64> %shl, %A
  55   ret <2 x i64> %add
  56 ; CHECK-LABEL: @test6(
  57 ; CHECK-NEXT: %add = mul <2 x i64> %A, <i64 5, i64 9>
  58 ; CHECK-NEXT: ret <2 x i64> %add
  59 }
  60
  61 define <2 x i64> @test7(<2 x i64> %A) {
  62   %shl = shl <2 x i64> %A, <i64 2, i64 3>
  63   %mul = mul <2 x i64> %A, <i64 3, i64 4>
  64   %add = add <2 x i64> %shl, %mul
  65   ret <2 x i64> %add
  66 ; CHECK-LABEL: @test7(
  67 ; CHECK-NEXT: %add = mul <2 x i64> %A, <i64 7, i64 12>
  68 ; CHECK-NEXT: ret <2 x i64> %add
  69 }
  70
  71 define <2 x i64> @test8(<2 x i64> %A) {
  72   %xor = xor <2 x i64> %A, <i64 -1, i64 -1>
  73   %add = add <2 x i64> %xor, <i64 2, i64 3>
  74   ret <2 x i64> %add
  75 ; CHECK-LABEL: @test8(
  76 ; CHECK-NEXT: %add = sub <2 x i64> <i64 1, i64 2>, %A
  77 ; CHECK-NEXT: ret <2 x i64> %add
  78 }
  79
  80 define i16 @test9(i16 %a) {
  81        %b = mul i16 %a, 2
  82        %c = mul i16 %a, 32767
  83        %d = add i16 %b, %c
  84        ret i16 %d
  85 ; CHECK-LABEL: @test9(
  86 ; CHECK-NEXT:  %d = mul i16 %a, -32767
  87 ; CHECK-NEXT:  ret i16 %d
  88 }
  89
  90 ; y + (~((x >> 3) & 0x55555555) + 1) -> y - ((x >> 3) & 0x55555555)
  91 define i32 @test10(i32 %x, i32 %y) {
  92   %shr = ashr i32 %x, 3
  93   %shr.not = or i32 %shr, -1431655766
  94   %neg = xor i32 %shr.not, 1431655765
  95   %add = add i32 %y, 1
  96   %add1 = add i32 %add, %neg
  97   ret i32 %add1
  98 ; CHECK-LABEL: @test10(
  99 ; CHECK-NEXT: [[SHR:%[a-z0-9]+]] = ashr i32 %x, 3
 100 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = and i32 [[SHR]], 1431655765
 101 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 102 ; CHECK-NEXT: ret i32 [[SUB]]
 103 }
 104
 105 ; y + (~(x & 0x55555555) + 1) -> y - (x & 0x55555555)
 106 define i32 @test11(i32 %x, i32 %y) {
 107   %x.not = or i32 %x, -1431655766
 108   %neg = xor i32 %x.not, 1431655765
 109   %add = add i32 %y, 1
 110   %add1 = add i32 %add, %neg
 111   ret i32 %add1
 112 ; CHECK-LABEL: @test11(
 113 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = and i32 %x, 1431655765
 114 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 115 ; CHECK-NEXT: ret i32 [[SUB]]
 116 }
 117
 118 ; (y + 1) + ~(x & 0x55555555) -> y - (x & 0x55555555)
 119 define i32 @test12(i32 %x, i32 %y) {
 120   %add = add nsw i32 %y, 1
 121   %x.not = or i32 %x, -1431655766
 122   %neg = xor i32 %x.not, 1431655765
 123   %add1 = add nsw i32 %add, %neg
 124   ret i32 %add1
 125 ; CHECK-LABEL: @test12(
 126 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = and i32 %x, 1431655765
 127 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 128 ; CHECK-NEXT: ret i32 [[SUB]]
 129 }
 130
 131 ; y + (~(x & 0x55555556) + 1) -> y - (x & 0x55555556)
 132 define i32 @test13(i32 %x, i32 %y) {
 133   %x.not = or i32 %x, -1431655767
 134   %neg = xor i32 %x.not, 1431655766
 135   %add = add i32 %y, 1
 136   %add1 = add i32 %add, %neg
 137   ret i32 %add1
 138 ; CHECK-LABEL: @test13(
 139 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = and i32 %x, 1431655766
 140 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 141 ; CHECK-NEXT: ret i32 [[SUB]]
 142 }
 143
 144 ; (y + 1) + ~(x & 0x55555556) -> y - (x & 0x55555556)
 145 define i32 @test14(i32 %x, i32 %y) {
 146   %add = add nsw i32 %y, 1
 147   %x.not = or i32 %x, -1431655767
 148   %neg = xor i32 %x.not, 1431655766
 149   %add1 = add nsw i32 %add, %neg
 150   ret i32 %add1
 151 ; CHECK-LABEL: @test14(
 152 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = and i32 %x, 1431655766
 153 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 154 ; CHECK-NEXT: ret i32 [[SUB]]
 155 }
 156
 157 ; y + (~(x | 0x55555556) + 1) -> y - (x | 0x55555556)
 158 define i32 @test15(i32 %x, i32 %y) {
 159   %x.not = and i32 %x, -1431655767
 160   %neg = xor i32 %x.not, -1431655767
 161   %add = add i32 %y, 1
 162   %add1 = add i32 %add, %neg
 163   ret i32 %add1
 164 ; CHECK-LABEL: @test15(
 165 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = or i32 %x, 1431655766
 166 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 167 ; CHECK-NEXT: ret i32 [[SUB]]
 168 }
 169
 170 ; (y + 1) + ~(x | 0x55555556) -> y - (x | 0x555555556)
 171 define i32 @test16(i32 %x, i32 %y) {
 172   %add = add nsw i32 %y, 1
 173   %x.not = and i32 %x, -1431655767
 174   %neg = xor i32 %x.not, -1431655767
 175   %add1 = add nsw i32 %add, %neg
 176   ret i32 %add1
 177 ; CHECK-LABEL: @test16(
 178 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = or i32 %x, 1431655766
 179 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 180 ; CHECK-NEXT: ret i32 [[SUB]]
 181 }
 182
 183 ; y + (~(x | 0x55555555) + 1) -> y - (x | 0x55555555)
 184 define i32 @test17(i32 %x, i32 %y) {
 185   %x.not = and i32 %x, -1431655766
 186   %add2 = xor i32 %x.not, -1431655765
 187   %add1 = add nsw i32 %add2, %y
 188   ret i32 %add1
 189 ; CHECK-LABEL: @test17(
 190 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = or i32 %x, 1431655765
 191 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 192 ; CHECK-NEXT: ret i32 [[SUB]]
 193 }
 194
 195 ; (y + 1) + ~(x | 0x55555555) -> y - (x | 0x55555555)
 196 define i32 @test18(i32 %x, i32 %y) {
 197   %add = add nsw i32 %y, 1
 198   %x.not = and i32 %x, -1431655766
 199   %neg = xor i32 %x.not, -1431655766
 200   %add1 = add nsw i32 %add, %neg
 201   ret i32 %add1
 202 ; CHECK-LABEL: @test18(
 203 ; CHECK-NEXT: [[AND:%[a-z0-9]+]] = or i32 %x, 1431655765
 204 ; CHECK-NEXT: [[SUB:%[a-z0-9]+]] = sub i32 %y, [[AND]]
 205 ; CHECK-NEXT: ret i32 [[SUB]]
 206 }
 207
 208 define i16 @add_nsw_mul_nsw(i16 %x) {
 209  %add1 = add nsw i16 %x, %x
 210  %add2 = add nsw i16 %add1, %x
 211  ret i16 %add2
 212 ; CHECK-LABEL: @add_nsw_mul_nsw(
 213 ; CHECK-NEXT: %add2 = mul nsw i16 %x, 3
 214 ; CHECK-NEXT: ret i16 %add2
 215 }
 216
 217 define i16 @mul_add_to_mul_1(i16 %x) {
 218  %mul1 = mul nsw i16 %x, 8
 219  %add2 = add nsw i16 %x, %mul1
 220  ret i16 %add2
 221 ; CHECK-LABEL: @mul_add_to_mul_1(
 222 ; CHECK-NEXT: %add2 = mul nsw i16 %x, 9
 223 ; CHECK-NEXT: ret i16 %add2
 224 }
 225
 226 define i16 @mul_add_to_mul_2(i16 %x) {
 227  %mul1 = mul nsw i16 %x, 8
 228  %add2 = add nsw i16 %mul1, %x
 229  ret i16 %add2
 230 ; CHECK-LABEL: @mul_add_to_mul_2(
 231 ; CHECK-NEXT: %add2 = mul nsw i16 %x, 9
 232 ; CHECK-NEXT: ret i16 %add2
 233 }
 234
 235 define i16 @mul_add_to_mul_3(i16 %a) {
 236  %mul1 = mul i16 %a, 2
 237  %mul2 = mul i16 %a, 3
 238  %add = add nsw i16 %mul1, %mul2
 239  ret i16 %add
 240 ; CHECK-LABEL: @mul_add_to_mul_3(
 241 ; CHECK-NEXT: %add = mul i16 %a, 5
 242 ; CHECK-NEXT: ret i16 %add
 243 }
 244
 245 define i16 @mul_add_to_mul_4(i16 %a) {
 246  %mul1 = mul nsw i16 %a, 2
 247  %mul2 = mul nsw i16 %a, 7
 248  %add = add nsw i16 %mul1, %mul2
 249  ret i16 %add
 250 ; CHECK-LABEL: @mul_add_to_mul_4(
 251 ; CHECK-NEXT: %add = mul nsw i16 %a, 9
 252 ; CHECK-NEXT: ret i16 %add
 253 }
 254
 255 define i16 @mul_add_to_mul_5(i16 %a) {
 256  %mul1 = mul nsw i16 %a, 3
 257  %mul2 = mul nsw i16 %a, 7
 258  %add = add nsw i16 %mul1, %mul2
 259  ret i16 %add
 260 ; CHECK-LABEL: @mul_add_to_mul_5(
 261 ; CHECK-NEXT: %add = mul nsw i16 %a, 10
 262 ; CHECK-NEXT: ret i16 %add
 263 }
 264
 265 define i32 @mul_add_to_mul_6(i32 %x, i32 %y) {
 266   %mul1 = mul nsw i32 %x, %y
 267   %mul2 = mul nsw i32 %mul1, 5
 268   %add = add nsw i32 %mul1, %mul2
 269   ret i32 %add
 270 ; CHECK-LABEL: @mul_add_to_mul_6(
 271 ; CHECK-NEXT: %mul1 = mul nsw i32 %x, %y
 272 ; CHECK-NEXT: %add = mul nsw i32 %mul1, 6
 273 ; CHECK-NEXT: ret i32 %add
 274 }
 275
 276 ; This test and the next test verify that when a range metadata is attached to
 277 ; llvm.cttz, ValueTracking correctly intersects the range specified by the
 278 ; metadata and the range implied by the intrinsic.
 279 ;
 280 ; In this test, the range specified by the metadata is more strict. Therefore,
 281 ; ValueTracking uses that range.
 282 define i16 @add_cttz(i16 %a) {
 283 ; CHECK-LABEL: @add_cttz(
 284   ; llvm.cttz.i16(..., /*is_zero_undefined=*/true) implies the value returned
 285   ; is in [0, 16). The range metadata indicates the value returned is in [0, 8).
 286   ; Intersecting these ranges, we know the value returned is in [0, 8).
 287   ; Therefore, InstCombine will transform
 288   ;     add %cttz, 1111 1111 1111 1000 ; decimal -8
 289   ; to
 290   ;     or  %cttz, 1111 1111 1111 1000
 291   %cttz = call i16 @llvm.cttz.i16(i16 %a, i1 true), !range !0
 292   %b = add i16 %cttz, -8
 293 ; CHECK: or i16 %cttz, -8
 294   ret i16 %b
 295 }
 296 declare i16 @llvm.cttz.i16(i16, i1)
 297 !0 = !{i16 0, i16 8}
 298
 299 ; Similar to @add_cttz, but in this test, the range implied by the
 300 ; intrinsic is more strict. Therefore, ValueTracking uses that range.
 301 define i16 @add_cttz_2(i16 %a) {
 302 ; CHECK-LABEL: @add_cttz_2(
 303   ; llvm.cttz.i16(..., /*is_zero_undefined=*/true) implies the value returned
 304   ; is in [0, 16). The range metadata indicates the value returned is in
 305   ; [0, 32). Intersecting these ranges, we know the value returned is in
 306   ; [0, 16). Therefore, InstCombine will transform
 307   ;     add %cttz, 1111 1111 1111 0000 ; decimal -16
 308   ; to
 309   ;     or  %cttz, 1111 1111 1111 0000
 310   %cttz = call i16 @llvm.cttz.i16(i16 %a, i1 true), !range !1
 311   %b = add i16 %cttz, -16
 312 ; CHECK: or i16 %cttz, -16
 313   ret i16 %b
 314 }
 315 !1 = !{i16 0, i16 32}
 316
 317 define i32 @add_or_and(i32 %x, i32 %y) {
 318   %or = or i32 %x, %y
 319   %and = and i32 %x, %y
 320   %add = add i32 %or, %and
 321   ret i32 %add
 322 ; CHECK-LABEL: @add_or_and(
 323 ; CHECK-NEXT: add i32 %x, %y
 324 ; CHECK-NEXT: ret i32
 325 }
 326
 327 define i32 @add_nsw_or_and(i32 %x, i32 %y) {
 328   %or = or i32 %x, %y
 329   %and = and i32 %x, %y
 330   %add = add nsw i32 %or, %and
 331   ret i32 %add
 332 ; CHECK-LABEL: @add_nsw_or_and(
 333 ; CHECK-NEXT: add nsw i32 %x, %y
 334 ; CHECK-NEXT: ret i32
 335 }
 336
 337 define i32 @add_nuw_or_and(i32 %x, i32 %y) {
 338   %or = or i32 %x, %y
 339   %and = and i32 %x, %y
 340   %add = add nuw i32 %or, %and
 341   ret i32 %add
 342 ; CHECK-LABEL: @add_nuw_or_and(
 343 ; CHECK-NEXT: add nuw i32 %x, %y
 344 ; CHECK-NEXT: ret i32
 345 }
 346
 347 define i32 @add_nuw_nsw_or_and(i32 %x, i32 %y) {
 348   %or = or i32 %x, %y
 349   %and = and i32 %x, %y
 350   %add = add nsw nuw i32 %or, %and
 351   ret i32 %add
 352 ; CHECK-LABEL: @add_nuw_nsw_or_and(
 353 ; CHECK-NEXT: add nuw nsw i32 %x, %y
 354 ; CHECK-NEXT: ret i32
 355 }