xref: /llvm-project/llvm/test/CodeGen/X86/switch-edge-weight.ll (revision 9cac4e6d1403554b06ec2fc9d834087b1234b695)
1; RUN: llc -mtriple=x86_64-- -print-after=finalize-isel %s -o /dev/null 2>&1 | FileCheck %s
2
3declare void @foo(i32)
4
5; CHECK-LABEL: test
6
7define void @test(i32 %x) nounwind {
8entry:
9  switch i32 %x, label %sw.default [
10    i32 1, label %sw.bb
11    i32 155, label %sw.bb
12    i32 156, label %sw.bb
13    i32 157, label %sw.bb
14    i32 158, label %sw.bb
15    i32 159, label %sw.bb
16    i32 1134, label %sw.bb
17    i32 1140, label %sw.bb
18  ], !prof !1
19
20sw.bb:
21  call void @foo(i32 0)
22  br label %sw.epilog
23
24sw.default:
25  call void @foo(i32 1)
26  br label %sw.epilog
27
28sw.epilog:
29  ret void
30
31; Check if weights are correctly assigned to edges generated from switch
32; statement.
33;
34; CHECK: bb.0{{[0-9a-zA-Z.]*}}:
35; %bb.0 to %bb.4: [0, 1133] (65 = 60 + 5)
36; %bb.0 to %bb.5: [1134, UINT32_MAX] (25 = 20 + 5)
37; CHECK: successors: %bb.4(0x5c71c71c), %bb.5(0x238e38e4)
38;
39; CHECK: bb.4{{[0-9a-zA-Z.]*}}:
40; %bb.4 to %bb.1: [155, 159] (50)
41; %bb.4 to %bb.5: [0, 1133] - [155, 159] (15 = 10 + 5)
42; CHECK: successors: %bb.1(0x62762762), %bb.7(0x1d89d89e)
43;
44; CHECK: bb.5{{[0-9a-zA-Z.]*}}:
45; %bb.5 to %bb.1: {1140} (10)
46; %bb.5 to %bb.6: [1134, UINT32_MAX] - {1140} (15 = 10 + 5)
47; CHECK: successors: %bb.1(0x33333333), %bb.6(0x4ccccccd)
48;
49; CHECK: bb.6{{[0-9a-zA-Z.]*}}:
50; %bb.6 to %bb.1: {1134} (10)
51; %bb.6 to %bb.2: [1134, UINT32_MAX] - {1134, 1140} (5)
52; CHECK: successors: %bb.1(0x55555555), %bb.2(0x2aaaaaab)
53}
54
55; CHECK-LABEL: test2
56
57define void @test2(i32 %x) nounwind {
58entry:
59
60; In this switch statement, there is an edge from jump table to default
61; statement.
62
63  switch i32 %x, label %sw.default [
64    i32 1, label %sw.bb
65    i32 10, label %sw.bb2
66    i32 11, label %sw.bb3
67    i32 12, label %sw.bb4
68    i32 13, label %sw.bb5
69    i32 14, label %sw.bb5
70  ], !prof !3
71
72sw.bb:
73  call void @foo(i32 0)
74  br label %sw.epilog
75
76sw.bb2:
77  call void @foo(i32 2)
78  br label %sw.epilog
79
80sw.bb3:
81  call void @foo(i32 3)
82  br label %sw.epilog
83
84sw.bb4:
85  call void @foo(i32 4)
86  br label %sw.epilog
87
88sw.bb5:
89  call void @foo(i32 5)
90  br label %sw.epilog
91
92sw.default:
93  call void @foo(i32 1)
94  br label %sw.epilog
95
96sw.epilog:
97  ret void
98
99; Check if weights are correctly assigned to edges generated from switch
100; statement.
101;
102; CHECK: bb.0{{[0-9a-zA-Z.]*}}:
103; %bb.0 to %bb.6: {0} + [15, UINT32_MAX] (5)
104; %bb.0 to %bb.8: [1, 14] (jump table) (65 = 60 + 5)
105; CHECK: successors: %bb.6(0x09249249), %bb.8(0x76db6db7)
106;
107; CHECK: bb.8{{[0-9a-zA-Z.]*}}:
108; %bb.8 to %bb.1: {1} (10)
109; %bb.8 to %bb.6: [2, 9] (5)
110; %bb.8 to %bb.2: {10} (10)
111; %bb.8 to %bb.3: {11} (10)
112; %bb.8 to %bb.4: {12} (10)
113; %bb.8 to %bb.5: {13, 14} (20)
114; CHECK: successors: %bb.1(0x13b13b14), %bb.6(0x09d89d8a), %bb.2(0x13b13b14), %bb.3(0x13b13b14), %bb.4(0x13b13b14), %bb.5(0x27627628)
115}
116
117; CHECK-LABEL: test3
118
119define void @test3(i32 %x) nounwind {
120entry:
121
122; In this switch statement, there is no edge from jump table to default
123; statement.
124
125  switch i32 %x, label %sw.default [
126    i32 10, label %sw.bb
127    i32 11, label %sw.bb2
128    i32 12, label %sw.bb3
129    i32 13, label %sw.bb4
130    i32 14, label %sw.bb5
131  ], !prof !2
132
133sw.bb:
134  call void @foo(i32 0)
135  br label %sw.epilog
136
137sw.bb2:
138  call void @foo(i32 2)
139  br label %sw.epilog
140
141sw.bb3:
142  call void @foo(i32 3)
143  br label %sw.epilog
144
145sw.bb4:
146  call void @foo(i32 4)
147  br label %sw.epilog
148
149sw.bb5:
150  call void @foo(i32 5)
151  br label %sw.epilog
152
153sw.default:
154  call void @foo(i32 1)
155  br label %sw.epilog
156
157sw.epilog:
158  ret void
159
160; Check if weights are correctly assigned to edges generated from switch
161; statement.
162;
163; CHECK: bb.0{{[0-9a-zA-Z.]*}}:
164; %bb.0 to %bb.6: [0, 9] + [15, UINT32_MAX] {10}
165; %bb.0 to %bb.8: [10, 14] (jump table) (50)
166; CHECK: successors: %bb.6(0x15555555), %bb.8(0x6aaaaaab)
167;
168; CHECK: bb.8{{[0-9a-zA-Z.]*}}:
169; %bb.8 to %bb.1: {10} (10)
170; %bb.8 to %bb.2: {11} (10)
171; %bb.8 to %bb.3: {12} (10)
172; %bb.8 to %bb.4: {13} (10)
173; %bb.8 to %bb.5: {14} (10)
174; CHECK: successors: %bb.1(0x1999999a), %bb.2(0x1999999a), %bb.3(0x1999999a), %bb.4(0x1999999a), %bb.5(0x1999999a)
175}
176
177; CHECK-LABEL: test4
178
179define void @test4(i32 %x) nounwind {
180entry:
181
182; In this switch statement, there is no edge from bit test to default basic
183; block.
184
185  switch i32 %x, label %sw.default [
186    i32 1, label %sw.bb
187    i32 111, label %sw.bb2
188    i32 112, label %sw.bb3
189    i32 113, label %sw.bb3
190    i32 114, label %sw.bb2
191    i32 115, label %sw.bb2
192  ], !prof !3
193
194sw.bb:
195  call void @foo(i32 0)
196  br label %sw.epilog
197
198sw.bb2:
199  call void @foo(i32 2)
200  br label %sw.epilog
201
202sw.bb3:
203  call void @foo(i32 3)
204  br label %sw.epilog
205
206sw.default:
207  call void @foo(i32 1)
208  br label %sw.epilog
209
210sw.epilog:
211  ret void
212
213; Check if weights are correctly assigned to edges generated from switch
214; statement.
215;
216; CHECK: bb.0{{[0-9a-zA-Z.]*}}:
217; %bb.0 to %bb.6: [0, 110] + [116, UINT32_MAX] (20)
218; %bb.0 to %bb.7: [111, 115] (bit test) (50)
219; CHECK: successors: %bb.6(0x24924925), %bb.7(0x5b6db6db)
220;
221; CHECK: bb.7{{[0-9a-zA-Z.]*}}:
222; %bb.7 to %bb.2: {111, 114, 115} (30)
223; %bb.7 to %bb.3: {112, 113} (20)
224; CHECK: successors: %bb.2(0x4ccccccd), %bb.3(0x33333333)
225}
226
227; CHECK-LABEL: test5
228
229define void @test5(i32 %x) nounwind {
230entry:
231
232; In this switch statement, there is an edge from jump table to default basic
233; block.
234
235  switch i32 %x, label %sw.default [
236    i32 4, label %sw.bb
237    i32 20, label %sw.bb2
238    i32 28, label %sw.bb3
239    i32 36, label %sw.bb4
240    i32 124, label %sw.bb5
241  ], !prof !2
242
243sw.bb:
244  call void @foo(i32 0)
245  br label %sw.epilog
246
247sw.bb2:
248  call void @foo(i32 1)
249  br label %sw.epilog
250
251sw.bb3:
252  call void @foo(i32 2)
253  br label %sw.epilog
254
255sw.bb4:
256  call void @foo(i32 3)
257  br label %sw.epilog
258
259sw.bb5:
260  call void @foo(i32 4)
261  br label %sw.epilog
262
263sw.default:
264  call void @foo(i32 5)
265  br label %sw.epilog
266
267sw.epilog:
268  ret void
269
270; Check if weights are correctly assigned to edges generated from switch
271; statement.
272;
273; CHECK: bb.0{{[0-9a-zA-Z.]*}}:
274; %bb.0 to %bb.6: [10, UINT32_MAX] (15)
275; %bb.0 to %bb.8: [4, 20, 28, 36] (jump table) (45)
276; CHECK: successors: %bb.8(0x20000001), %bb.9(0x5fffffff)
277}
278
279!1 = !{!"branch_weights", i32 10, i32 10, i32 10, i32 10, i32 10, i32 10, i32 10, i32 10, i32 10}
280!2 = !{!"branch_weights", i32 10, i32 10, i32 10, i32 10, i32 10, i32 10}
281!3 = !{!"branch_weights", i32 10, i32 10, i32 10, i32 10, i32 10, i32 10, i32 10}
282