xref: /llvm-project/llvm/test/Transforms/Reassociate/negation.ll (revision 38fffa630ee80163dc65e759392ad29798905679)
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
6define 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
21define 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
34define <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
49define <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
65define <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}
80