llvm/test/Transforms/Reassociate/negation.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=reassociate -S | FileCheck %s
   3
   4 ; Test that we can turn things like X*-(Y*Z) -> X*-1*Y*Z.
   5
   6 define i32 @test1(i32 %a, i32 %b, i32 %z) {
   7 ; CHECK-LABEL: @test1(
   8 ; CHECK-NEXT:    [[E:%.*]] = mul i32 [[A:%.*]], 12345
   9 ; CHECK-NEXT:    [[F:%.*]] = mul i32 [[E]], [[B:%.*]]
  10 ; CHECK-NEXT:    [[G:%.*]] = mul i32 [[F]], [[Z:%.*]]
  11 ; CHECK-NEXT:    ret i32 [[G]]
  12 ;
  13   %c = sub i32 0, %z
  14   %d = mul i32 %a, %b
  15   %e = mul i32 %c, %d
  16   %f = mul i32 %e, 12345
  17   %g = sub i32 0, %f
  18   ret i32 %g
  19 }
  20
  21 define i32 @test2(i32 %a, i32 %b, i32 %z) {
  22 ; CHECK-LABEL: @test2(
  23 ; CHECK-NEXT:    [[E:%.*]] = mul i32 [[A:%.*]], 40
  24 ; CHECK-NEXT:    [[F:%.*]] = mul i32 [[E]], [[Z:%.*]]
  25 ; CHECK-NEXT:    ret i32 [[F]]
  26 ;
  27   %d = mul i32 %z, 40
  28   %c = sub i32 0, %d
  29   %e = mul i32 %a, %c
  30   %f = sub i32 0, %e
  31   ret i32 %f
  32 }
  33
  34 define <2 x i32> @negate_vec_poisons(<2 x i32> %a, <2 x i32> %b, <2 x i32> %z) {
  35 ; CHECK-LABEL: @negate_vec_poisons(
  36 ; CHECK-NEXT:    [[E:%.*]] = mul <2 x i32> [[A:%.*]], splat (i32 40)
  37 ; CHECK-NEXT:    [[F:%.*]] = mul <2 x i32> [[E]], [[Z:%.*]]
  38 ; CHECK-NEXT:    ret <2 x i32> [[F]]
  39 ;
  40   %d = mul <2 x i32> %z, <i32 40, i32 40>
  41   %c = sub <2 x i32> <i32 0, i32 poison>, %d
  42   %e = mul <2 x i32> %a, %c
  43   %f = sub <2 x i32> <i32 0, i32 poison>, %e
  44   ret <2 x i32> %f
  45 }
  46
  47 ; Replacing %x with a partial undef negation is a miscompile.
  48
  49 define <2 x i32> @PR57683(<2 x i32> %x) {
  50 ; CHECK-LABEL: @PR57683(
  51 ; CHECK-NEXT:    [[PARTIAL_NEG:%.*]] = sub <2 x i32> <i32 poison, i32 0>, [[X:%.*]]
  52 ; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <2 x i32> [[PARTIAL_NEG]], <2 x i32> [[X]], <2 x i32> <i32 1, i32 3>
  53 ; CHECK-NEXT:    [[X_NEG:%.*]] = sub <2 x i32> zeroinitializer, [[X]]
  54 ; CHECK-NEXT:    [[SUB:%.*]] = add <2 x i32> [[X_NEG]], splat (i32 1)
  55 ; CHECK-NEXT:    [[R:%.*]] = add <2 x i32> [[SUB]], [[SHUF]]
  56 ; CHECK-NEXT:    ret <2 x i32> [[R]]
  57 ;
  58   %partial_neg = sub <2 x i32> <i32 poison, i32 0>, %x
  59   %shuf = shufflevector <2 x i32> %partial_neg, <2 x i32> %x, <2 x i32> <i32 1, i32 3>
  60   %sub = sub <2 x i32> <i32 1, i32 1>, %x
  61   %r = add <2 x i32> %sub, %shuf
  62   ret <2 x i32> %r
  63 }
  64
  65 define <2 x float> @PR57683_FP(<2 x float> %x) {
  66 ; CHECK-LABEL: @PR57683_FP(
  67 ; CHECK-NEXT:    [[PARTIAL_NEG:%.*]] = fsub reassoc nsz <2 x float> <float poison, float 0.000000e+00>, [[X:%.*]]
  68 ; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <2 x float> [[PARTIAL_NEG]], <2 x float> [[X]], <2 x i32> <i32 1, i32 3>
  69 ; CHECK-NEXT:    [[X_NEG:%.*]] = fneg reassoc nsz <2 x float> [[X]]
  70 ; CHECK-NEXT:    [[SUB:%.*]] = fadd reassoc nsz <2 x float> [[X_NEG]], splat (float 1.000000e+00)
  71 ; CHECK-NEXT:    [[R:%.*]] = fadd reassoc nsz <2 x float> [[SUB]], [[SHUF]]
  72 ; CHECK-NEXT:    ret <2 x float> [[R]]
  73 ;
  74   %partial_neg = fsub reassoc nsz <2 x float> <float poison, float 0.0>, %x
  75   %shuf = shufflevector <2 x float> %partial_neg, <2 x float> %x, <2 x i32> <i32 1, i32 3>
  76   %sub = fsub reassoc nsz <2 x float> <float 1.0, float 1.0>, %x
  77   %r = fadd reassoc nsz <2 x float> %sub, %shuf
  78   ret <2 x float> %r
  79 }