1; RUN: opt -codegenprepare -S < %s | FileCheck %s
2
3target triple = "x86_64-unknown-unknown"
4
5; Nothing to sink and convert here.
6
7define i32 @no_sink(double %a, double* %b, i32 %x, i32 %y)  {
8entry:
9  %load = load double, double* %b, align 8
10  %cmp = fcmp olt double %load, %a
11  %sel = select i1 %cmp, i32 %x, i32 %y
12  ret i32 %sel
13
14; CHECK-LABEL: @no_sink(
15; CHECK:    %load = load double, double* %b, align 8
16; CHECK:    %cmp = fcmp olt double %load, %a
17; CHECK:    %sel = select i1 %cmp, i32 %x, i32 %y
18; CHECK:    ret i32 %sel
19}
20
21
22; An 'fdiv' is expensive, so sink it rather than speculatively execute it.
23
24define float @fdiv_true_sink(float %a, float %b) {
25entry:
26  %div = fdiv float %a, %b
27  %cmp = fcmp ogt float %a, 1.0
28  %sel = select i1 %cmp, float %div, float 2.0
29  ret float %sel
30
31; CHECK-LABEL: @fdiv_true_sink(
32; CHECK:    %cmp = fcmp ogt float %a, 1.0
33; CHECK:    br i1 %cmp, label %select.true.sink, label %select.end
34; CHECK:  select.true.sink:
35; CHECK:    %div = fdiv float %a, %b
36; CHECK:    br label %select.end
37; CHECK:  select.end:
38; CHECK:    %sel = phi float [ %div, %select.true.sink ], [ 2.000000e+00, %entry ]
39; CHECK:    ret float %sel
40}
41
42define float @fdiv_false_sink(float %a, float %b) {
43entry:
44  %div = fdiv float %a, %b
45  %cmp = fcmp ogt float %a, 3.0
46  %sel = select i1 %cmp, float 4.0, float %div
47  ret float %sel
48
49; CHECK-LABEL: @fdiv_false_sink(
50; CHECK:    %cmp = fcmp ogt float %a, 3.0
51; CHECK:    br i1 %cmp, label %select.end, label %select.false.sink
52; CHECK:  select.false.sink:
53; CHECK:    %div = fdiv float %a, %b
54; CHECK:    br label %select.end
55; CHECK:  select.end:
56; CHECK:    %sel = phi float [ 4.000000e+00, %entry ], [ %div, %select.false.sink ]
57; CHECK:    ret float %sel
58}
59
60define float @fdiv_both_sink(float %a, float %b) {
61entry:
62  %div1 = fdiv float %a, %b
63  %div2 = fdiv float %b, %a
64  %cmp = fcmp ogt float %a, 5.0
65  %sel = select i1 %cmp, float %div1, float %div2
66  ret float %sel
67
68; CHECK-LABEL: @fdiv_both_sink(
69; CHECK:    %cmp = fcmp ogt float %a, 5.0
70; CHECK:    br i1 %cmp, label %select.true.sink, label %select.false.sink
71; CHECK:  select.true.sink:
72; CHECK:    %div1 = fdiv float %a, %b
73; CHECK:    br label %select.end
74; CHECK:  select.false.sink:
75; CHECK:    %div2 = fdiv float %b, %a
76; CHECK:    br label %select.end
77; CHECK:  select.end:
78; CHECK:    %sel = phi float [ %div1, %select.true.sink ], [ %div2, %select.false.sink ]
79; CHECK:    ret float %sel
80}
81
82; An 'fadd' is not too expensive, so it's ok to speculate.
83
84define float @fadd_no_sink(float %a, float %b) {
85  %add = fadd float %a, %b
86  %cmp = fcmp ogt float 6.0, %a
87  %sel = select i1 %cmp, float %add, float 7.0
88  ret float %sel
89
90; CHECK-LABEL: @fadd_no_sink(
91; CHECK:  %sel = select i1 %cmp, float %add, float 7.0
92}
93
94; Possible enhancement: sinkability is only calculated with the direct
95; operand of the select, so we don't try to sink this. The fdiv cost is not
96; taken into account.
97
98define float @fdiv_no_sink(float %a, float %b) {
99entry:
100  %div = fdiv float %a, %b
101  %add = fadd float %div, %b
102  %cmp = fcmp ogt float %a, 1.0
103  %sel = select i1 %cmp, float %add, float 8.0
104  ret float %sel
105
106; CHECK-LABEL: @fdiv_no_sink(
107; CHECK:  %sel = select i1 %cmp, float %add, float 8.0
108}
109
110; Do not transform the CFG if the select operands may have side effects.
111
112declare i64* @bar(i32, i32, i32)
113declare i64* @baz(i32, i32, i32)
114
115define i64* @calls_no_sink(i32 %in) {
116  %call1 = call i64* @bar(i32 1, i32 2, i32 3)
117  %call2 = call i64* @baz(i32 1, i32 2, i32 3)
118  %tobool = icmp ne i32 %in, 0
119  %sel = select i1 %tobool, i64* %call1, i64* %call2
120  ret i64* %sel
121
122; CHECK-LABEL: @calls_no_sink(
123; CHECK:  %sel = select i1 %tobool, i64* %call1, i64* %call2
124}
125
126define i32 @sdiv_no_sink(i32 %a, i32 %b) {
127  %div1 = sdiv i32 %a, %b
128  %div2 = sdiv i32 %b, %a
129  %cmp = icmp sgt i32 %a, 5
130  %sel = select i1 %cmp, i32 %div1, i32 %div2
131  ret i32 %sel
132
133; CHECK-LABEL: @sdiv_no_sink(
134; CHECK:  %sel = select i1 %cmp, i32 %div1, i32 %div2
135}
136
137