1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt < %s -instcombine -S | FileCheck %s
3
4target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
5
6; Tests folding constants from two similar selects that feed a add
7
8define float @test1a(i1 zeroext %arg) #0 {
9; CHECK-LABEL: @test1a(
10; CHECK-NEXT:    [[TMP2:%.*]] = select i1 [[ARG:%.*]], float 6.000000e+00, float 1.500000e+01
11; CHECK-NEXT:    ret float [[TMP2]]
12;
13  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
14  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
15  %tmp2 = fadd float %tmp, %tmp1
16  ret float %tmp2
17}
18
19; Tests folding multiple expression constants from similar selects that feed a adds
20
21define float @test1b(i1 zeroext %arg) #0 {
22; CHECK-LABEL: @test1b(
23; CHECK-NEXT:    [[TMP5:%.*]] = select i1 [[ARG:%.*]], float 7.250000e+00, float 2.800000e+01
24; CHECK-NEXT:    ret float [[TMP5]]
25;
26  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
27  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
28  %tmp2 = select i1 %arg, float 2.500000e-01, float 4.000000e+00
29  %tmp3 = fadd float %tmp, %tmp1
30  %tmp4 = fadd float %tmp2, %tmp1
31  %tmp5 = fadd float %tmp4, %tmp3
32  ret float %tmp5
33}
34
35; Tests folding constants from two similar selects that feed a sub
36
37define float @test2(i1 zeroext %arg) #0 {
38; CHECK-LABEL: @test2(
39; CHECK-NEXT:    [[TMP2:%.*]] = select i1 [[ARG:%.*]], float 4.000000e+00, float -3.000000e+00
40; CHECK-NEXT:    ret float [[TMP2]]
41;
42  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
43  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
44  %tmp2 = fsub float %tmp, %tmp1
45  ret float %tmp2
46}
47
48; Tests folding constants from two similar selects that feed a mul
49
50define float @test3(i1 zeroext %arg) #0 {
51; CHECK-LABEL: @test3(
52; CHECK-NEXT:    [[TMP2:%.*]] = select i1 [[ARG:%.*]], float 5.000000e+00, float 5.400000e+01
53; CHECK-NEXT:    ret float [[TMP2]]
54;
55  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
56  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
57  %tmp2 = fmul float %tmp, %tmp1
58  ret float %tmp2
59}
60
61declare void @use_float(float)
62
63; Tests folding constants if the selects have multiple uses but
64; we can fold away the binary op with a select.
65
66define float @test4(i1 zeroext %arg) #0 {
67; CHECK-LABEL: @test4(
68; CHECK-NEXT:    [[TMP:%.*]] = select i1 [[ARG:%.*]], float 5.000000e+00, float 6.000000e+00
69; CHECK-NEXT:    [[TMP2:%.*]] = select i1 [[ARG]], float 5.000000e+00, float 5.400000e+01
70; CHECK-NEXT:    call void @use_float(float [[TMP]])
71; CHECK-NEXT:    ret float [[TMP2]]
72;
73  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
74  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
75  %tmp2 = fmul float %tmp, %tmp1
76  call void @use_float(float %tmp)
77  ret float %tmp2
78}
79
80; Tests not folding constants if we cannot fold away any of the selects.
81
82define float @test5(i1 zeroext %arg, float %div) {
83; CHECK-LABEL: @test5(
84; CHECK-NEXT:    [[TMP:%.*]] = select i1 [[ARG:%.*]], float [[DIV:%.*]], float 5.000000e+00
85; CHECK-NEXT:    [[MUL:%.*]] = fmul contract float [[TMP]], [[TMP]]
86; CHECK-NEXT:    call void @use_float(float [[TMP]])
87; CHECK-NEXT:    ret float [[MUL]]
88;
89  %tmp = select i1 %arg, float %div, float 5.000000e+00
90  %mul = fmul contract float %tmp, %tmp
91  call void @use_float(float %tmp)
92  ret float %mul
93}
94
95define float @fmul_nnan_nsz(i1 %cond, float %val) {
96; CHECK-LABEL: @fmul_nnan_nsz(
97; CHECK-NEXT:    [[LHS:%.*]] = select i1 [[COND:%.*]], float [[VAL:%.*]], float 0.000000e+00
98; CHECK-NEXT:    [[RHS:%.*]] = select i1 [[COND]], float -0.000000e+00, float [[VAL]]
99; CHECK-NEXT:    [[MUL:%.*]] = fmul nnan nsz float [[LHS]], [[RHS]]
100; CHECK-NEXT:    ret float [[MUL]]
101;
102  %lhs = select i1 %cond, float %val, float +0.0
103  %rhs = select i1 %cond, float -0.0, float %val
104  %mul = fmul nnan nsz float %lhs, %rhs
105  ret float %mul
106}
107
108define <2 x float> @fadd_nsz(<2 x i1> %cond, <2 x float> %val) {
109; CHECK-LABEL: @fadd_nsz(
110; CHECK-NEXT:    [[LHS:%.*]] = select <2 x i1> [[COND:%.*]], <2 x float> [[VAL:%.*]], <2 x float> zeroinitializer
111; CHECK-NEXT:    [[RHS:%.*]] = select <2 x i1> [[COND]], <2 x float> zeroinitializer, <2 x float> [[VAL]]
112; CHECK-NEXT:    [[ADD:%.*]] = fadd nsz <2 x float> [[LHS]], [[RHS]]
113; CHECK-NEXT:    ret <2 x float> [[ADD]]
114;
115  %lhs = select <2 x i1> %cond, <2 x float> %val, <2 x float> <float +0.0, float +0.0>
116  %rhs = select <2 x i1> %cond, <2 x float> <float +0.0, float +0.0>, <2 x float> %val
117  %add = fadd nsz <2 x float> %lhs, %rhs
118  ret <2 x float> %add
119}
120
121define double @fsub_nnan(i1 %cond, double %val, double %val2) {
122; CHECK-LABEL: @fsub_nnan(
123; CHECK-NEXT:    [[LHS:%.*]] = select i1 [[COND:%.*]], double [[VAL:%.*]], double [[VAL2:%.*]]
124; CHECK-NEXT:    [[RHS:%.*]] = select i1 [[COND]], double [[VAL]], double 7.000000e+00
125; CHECK-NEXT:    [[ADD:%.*]] = fsub nnan double [[LHS]], [[RHS]]
126; CHECK-NEXT:    ret double [[ADD]]
127;
128  %lhs = select i1 %cond, double %val, double %val2
129  %rhs = select i1 %cond, double %val, double 7.0
130  %add = fsub nnan double %lhs, %rhs
131  ret double %add
132}
133
134define double @fdiv_nnan_nsz(i1 %cond, double %val, double %val2) {
135; CHECK-LABEL: @fdiv_nnan_nsz(
136; CHECK-NEXT:    [[LHS:%.*]] = select i1 [[COND:%.*]], double [[VAL2:%.*]], double 0.000000e+00
137; CHECK-NEXT:    [[RHS:%.*]] = select i1 [[COND]], double 4.200000e+01, double [[VAL:%.*]]
138; CHECK-NEXT:    [[ADD:%.*]] = fdiv nnan nsz double [[LHS]], [[RHS]]
139; CHECK-NEXT:    ret double [[ADD]]
140;
141  %lhs = select i1 %cond, double %val2, double 0.0
142  %rhs = select i1 %cond, double 42.0, double %val
143  %add = fdiv nnan nsz double %lhs, %rhs
144  ret double %add
145}
146