1; RUN: opt < %s -instcombine -S | FileCheck %s
2
3target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
4
5; Tests folding constants from two similar selects that feed a add
6define float @test1a(i1 zeroext %arg) #0 {
7  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
8  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
9  %tmp2 = fadd float %tmp, %tmp1
10  ret float %tmp2
11; CHECK-LABEL: @test1a(
12; CHECK: %tmp2 = select i1 %arg, float 6.000000e+00, float 1.500000e+01
13; CHECK-NOT: fadd
14; CHECK: ret float %tmp2
15}
16
17; Tests folding multiple expression constants from similar selects that feed a adds
18define float @test1b(i1 zeroext %arg) #0 {
19  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
20  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
21  %tmp2 = select i1 %arg, float 2.500000e-01, float 4.000000e+00
22  %tmp3 = fadd float %tmp, %tmp1
23  %tmp4 = fadd float %tmp2, %tmp1
24  %tmp5 = fadd float %tmp4, %tmp3
25  ret float %tmp5
26; CHECK-LABEL: @test1b(
27; CHECK: %tmp5 = select i1 %arg, float 7.250000e+00, float 2.800000e+01
28; CHECK-NOT: fadd
29; CHECK: ret float %tmp5
30}
31
32; Tests folding constants from two similar selects that feed a sub
33define float @test2(i1 zeroext %arg) #0 {
34  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
35  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
36  %tmp2 = fsub float %tmp, %tmp1
37  ret float %tmp2
38; CHECK-LABEL: @test2(
39; CHECK: %tmp2 = select i1 %arg, float 4.000000e+00, float -3.000000e+00
40; CHECK-NOT: fsub
41; CHECK: ret float %tmp2
42}
43
44; Tests folding constants from two similar selects that feed a mul
45define float @test3(i1 zeroext %arg) #0 {
46  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
47  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
48  %tmp2 = fmul float %tmp, %tmp1
49  ret float %tmp2
50; CHECK-LABEL: @test3(
51; CHECK: %tmp2 = select i1 %arg, float 5.000000e+00, float 5.400000e+01
52; CHECK-NOT: fmul
53; CHECK: ret float %tmp2
54}
55
56declare void @use_float(float)
57
58; Tests folding constants if the selects have multiple uses but
59; we can fold away the binary op with a select.
60define float @test4(i1 zeroext %arg) #0 {
61  %tmp = select i1 %arg, float 5.000000e+00, float 6.000000e+00
62  %tmp1 = select i1 %arg, float 1.000000e+00, float 9.000000e+00
63  %tmp2 = fmul float %tmp, %tmp1
64  call void @use_float(float %tmp)
65  ret float %tmp2
66; CHECK-LABEL: @test4(
67; CHECK: select i1 %arg, float 5.000000e+00, float 6.000000e+00
68; CHECK-NEXT: select i1 %arg, float 5.000000e+00, float 5.400000e+01
69; CHECK-NEXT: call void @use_float(float %tmp)
70; CHECK: ret float %tmp2
71}
72
73; Tests not folding constants if we cannot fold away any of the selects.
74define float @test5(i1 zeroext %arg, float %div) {
75  %tmp = select i1 %arg, float %div, float 5.000000e+00
76  %mul = fmul contract float %tmp, %tmp
77  call void @use_float(float %tmp)
78  ret float %mul
79; CHECK-LABEL: @test5(
80; CHECK: [[TMP:%.*]] = select i1 %arg, float %div, float 5.000000e+00
81; CHECK-NEXT: [[MUL:%.*]] = fmul contract float [[TMP]], [[TMP]]
82; CHECK-NOT: fmul contract float %div, %div
83; CHECK: ret float [[MUL]]
84}
85