1 ; RUN: opt -simplifycfg -S -o - < %s | FileCheck %s
3 declare void @helper(i32)
5 define void @test1(i1 %a, i1 %b) {
8 br i1 %a, label %Y, label %X, !prof !0
9 ; CHECK: br i1 %or.cond, label %Z, label %Y, !prof !0
13 br i1 %c, label %Z, label %Y, !prof !1
16 call void @helper(i32 0)
20 call void @helper(i32 1)
24 define void @test2(i1 %a, i1 %b) {
25 ; CHECK-LABEL: @test2(
27 br i1 %a, label %X, label %Y, !prof !1
28 ; CHECK: br i1 %or.cond, label %Z, label %Y, !prof !1
33 br i1 %c, label %Z, label %Y, !prof !2
36 call void @helper(i32 0)
40 call void @helper(i32 1)
44 define void @test3(i1 %a, i1 %b) {
45 ; CHECK-LABEL: @test3(
48 br i1 %a, label %X, label %Y, !prof !1
52 br i1 %c, label %Z, label %Y
55 call void @helper(i32 0)
59 call void @helper(i32 1)
63 define void @test4(i1 %a, i1 %b) {
64 ; CHECK-LABEL: @test4(
67 br i1 %a, label %X, label %Y
71 br i1 %c, label %Z, label %Y, !prof !1
74 call void @helper(i32 0)
78 call void @helper(i32 1)
82 ;; test5 - The case where it jumps to the default target will be removed.
83 define void @test5(i32 %M, i32 %N) nounwind uwtable {
85 switch i32 %N, label %sw2 [
90 ; CHECK-LABEL: @test5(
91 ; CHECK: switch i32 %N, label %sw2 [
92 ; CHECK: i32 3, label %sw.bb1
93 ; CHECK: i32 2, label %sw.bb
97 call void @helper(i32 0)
101 call void @helper(i32 1)
105 call void @helper(i32 2)
112 ;; test6 - Some cases of the second switch are pruned during optimization.
113 ;; Then the second switch will be converted to a branch, finally, the first
114 ;; switch and the branch will be merged into a single switch.
115 define void @test6(i32 %M, i32 %N) nounwind uwtable {
117 switch i32 %N, label %sw2 [
122 ; CHECK-LABEL: @test6(
123 ; CHECK: switch i32 %N, label %sw.epilog
124 ; CHECK: i32 3, label %sw.bb1
125 ; CHECK: i32 2, label %sw.bb
126 ; CHECK: i32 4, label %sw.bb5
130 call void @helper(i32 0)
134 call void @helper(i32 1)
138 ;; Here "case 2" is invalidated since the default case of the first switch
139 ;; does not include "case 2".
140 switch i32 %N, label %sw.epilog [
146 call void @helper(i32 2)
150 call void @helper(i32 3)
157 ;; This test is based on test1 but swapped the targets of the second branch.
158 define void @test1_swap(i1 %a, i1 %b) {
159 ; CHECK-LABEL: @test1_swap(
161 br i1 %a, label %Y, label %X, !prof !0
162 ; CHECK: br i1 %or.cond, label %Y, label %Z, !prof !4
166 br i1 %c, label %Y, label %Z, !prof !1
169 call void @helper(i32 0)
173 call void @helper(i32 1)
177 define void @test7(i1 %a, i1 %b) {
178 ; CHECK-LABEL: @test7(
181 br i1 %a, label %Y, label %X, !prof !0
182 ; CHECK: br i1 %brmerge, label %Y, label %Z, !prof !5
185 br i1 %c, label %Y, label %Z, !prof !6
188 call void @helper(i32 0)
192 call void @helper(i32 1)
196 ; Test basic folding to a conditional branch.
197 define void @test8(i64 %x, i64 %y) nounwind {
198 ; CHECK-LABEL: @test8(
200 %lt = icmp slt i64 %x, %y
201 ; CHECK: br i1 %lt, label %a, label %b, !prof !6
202 %qux = select i1 %lt, i32 0, i32 2
203 switch i32 %qux, label %bees [
209 call void @helper(i32 0) nounwind
212 call void @helper(i32 1) nounwind
215 call void @helper(i32 2) nounwind
219 ; Test edge splitting when the default target has icmp and unconditinal
221 define i1 @test9(i32 %x, i32 %y) nounwind {
222 ; CHECK-LABEL: @test9(
224 switch i32 %x, label %bees [
229 ; CHECK: switch i32 %x, label %bees [
230 ; CHECK: i32 0, label %a
231 ; CHECK: i32 1, label %end
232 ; CHECK: i32 2, label %end
233 ; CHECK: i32 92, label %end
237 call void @helper(i32 0) nounwind
238 %reta = icmp slt i32 %x, %y
242 %tmp = icmp eq i32 %x, 92
247 ; CHECK: %ret = phi i1 [ true, %entry ], [ false, %bees ], [ true, %entry ], [ true, %entry ]
248 %ret = phi i1 [ true, %entry ], [%tmp, %bees], [true, %entry]
249 call void @helper(i32 2) nounwind
253 define void @test10(i32 %x) nounwind readnone ssp noredzone {
255 switch i32 %x, label %lor.rhs [
256 i32 2, label %lor.end
257 i32 1, label %lor.end
258 i32 3, label %lor.end
262 call void @helper(i32 1) nounwind
266 call void @helper(i32 0) nounwind
269 ; CHECK-LABEL: @test10(
270 ; CHECK: %x.off = add i32 %x, -1
271 ; CHECK: %switch = icmp ult i32 %x.off, 3
272 ; CHECK: br i1 %switch, label %lor.end, label %lor.rhs, !prof !8
275 ; Remove dead cases from the switch.
276 define void @test11(i32 %x) nounwind {
278 switch i32 %i, label %a [
282 ; CHECK-LABEL: @test11(
283 ; CHECK: %cond = icmp eq i32 %i, 24
284 ; CHECK: br i1 %cond, label %c, label %a, !prof !9
287 call void @helper(i32 0) nounwind
290 call void @helper(i32 1) nounwind
293 call void @helper(i32 2) nounwind
297 ;; test12 - Don't crash if the whole switch is removed
298 define void @test12(i32 %M, i32 %N) nounwind uwtable {
300 switch i32 %N, label %sw.bb [
303 ; CHECK-LABEL: @test12(
305 ; CHECK-NEXT: call void @helper
306 ; CHECK-NEXT: ret void
309 call void @helper(i32 0)
316 ;; If every case is dead, make sure they are all removed. This used to
317 ;; crash trying to merge the metadata.
318 define void @test13(i32 %x) nounwind {
321 switch i32 %i, label %a [
325 ; CHECK-LABEL: @test13(
327 ; CHECK-NEXT: call void @helper
328 ; CHECK-NEXT: ret void
331 call void @helper(i32 0) nounwind
334 call void @helper(i32 1) nounwind
337 call void @helper(i32 2) nounwind
341 ;; When folding branches to common destination, the updated branch weights
342 ;; can exceed uint32 by more than factor of 2. We should keep halving the
343 ;; weights until they can fit into uint32.
344 @max_regno = common global i32 0, align 4
345 define void @test14(i32* %old, i32 %final) {
346 ; CHECK-LABEL: @test14
347 ; CHECK: br i1 %or.cond, label %for.exit, label %for.inc, !prof !10
351 %i.1 = phi i32 [ %inc19, %for.inc ], [ 0, %for.cond ]
352 %bit.0 = phi i32 [ %shl, %for.inc ], [ 1, %for.cond ]
353 %tobool = icmp eq i32 %bit.0, 0
354 br i1 %tobool, label %for.exit, label %for.body3, !prof !10
356 %v3 = load i32* @max_regno, align 4
357 %cmp4 = icmp eq i32 %i.1, %v3
358 br i1 %cmp4, label %for.exit, label %for.inc, !prof !11
360 %shl = shl i32 %bit.0, 1
361 %inc19 = add nsw i32 %i.1, 1
367 !0 = metadata !{metadata !"branch_weights", i32 3, i32 5}
368 !1 = metadata !{metadata !"branch_weights", i32 1, i32 1}
369 !2 = metadata !{metadata !"branch_weights", i32 1, i32 2}
370 !3 = metadata !{metadata !"branch_weights", i32 4, i32 3, i32 2, i32 1}
371 !4 = metadata !{metadata !"branch_weights", i32 4, i32 3, i32 2, i32 1}
372 !5 = metadata !{metadata !"branch_weights", i32 7, i32 6, i32 5}
373 !6 = metadata !{metadata !"branch_weights", i32 1, i32 3}
374 !7 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8, i32 7}
375 !8 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8}
376 !9 = metadata !{metadata !"branch_weights", i32 7, i32 6}
377 !10 = metadata !{metadata !"branch_weights", i32 672646, i32 21604207}
378 !11 = metadata !{metadata !"branch_weights", i32 6960, i32 21597248}
380 ; CHECK: !0 = metadata !{metadata !"branch_weights", i32 5, i32 11}
381 ; CHECK: !1 = metadata !{metadata !"branch_weights", i32 1, i32 5}
382 ; CHECK: !2 = metadata !{metadata !"branch_weights", i32 7, i32 1, i32 2}
383 ; CHECK: !3 = metadata !{metadata !"branch_weights", i32 49, i32 12, i32 24, i32 35}
384 ; CHECK: !4 = metadata !{metadata !"branch_weights", i32 11, i32 5}
385 ; CHECK: !5 = metadata !{metadata !"branch_weights", i32 17, i32 15}
386 ; CHECK: !6 = metadata !{metadata !"branch_weights", i32 9, i32 7}
387 ; CHECK: !7 = metadata !{metadata !"branch_weights", i32 17, i32 9, i32 8, i32 7, i32 17}
388 ; CHECK: !8 = metadata !{metadata !"branch_weights", i32 24, i32 33}
389 ; CHECK: !9 = metadata !{metadata !"branch_weights", i32 8, i32 33}
390 ;; The false weight prints out as a negative integer here, but inside llvm, we
391 ;; treat the weight as an unsigned integer.
392 ; CHECK: !10 = metadata !{metadata !"branch_weights", i32 112017436, i32 -735157296}