xref: /llvm-project/llvm/test/Transforms/LoopUnroll/peel-branch-weights.ll (revision 5103ef64fe4f60cc0fd518b514c712f4b4c03d98)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals
2; RUN: opt < %s -S -passes=loop-unroll -unroll-force-peel-count=2 2>&1 | FileCheck %s
3; RUN: opt < %s -S -passes=loop-unroll -unroll-force-peel-count=2 -disable-advanced-peeling 2>&1 | FileCheck %s --check-prefix=DISABLEADV
4
5declare i32 @get.x()
6
7; Test branch weight update for terminator with multiple fallthrough and
8; multiple exit edges.
9define void @test() {
10; CHECK-LABEL: @test(
11; CHECK-NEXT:  entry:
12; CHECK-NEXT:    br label [[LOOP_PEEL_BEGIN:%.*]]
13; CHECK:       loop.peel.begin:
14; CHECK-NEXT:    br label [[LOOP_PEEL:%.*]]
15; CHECK:       loop.peel:
16; CHECK-NEXT:    [[X_PEEL:%.*]] = call i32 @get.x()
17; CHECK-NEXT:    switch i32 [[X_PEEL]], label [[LOOP_LATCH_PEEL:%.*]] [
18; CHECK-NEXT:    i32 0, label [[LOOP_LATCH_PEEL]]
19; CHECK-NEXT:    i32 1, label [[LOOP_EXIT:%.*]]
20; CHECK-NEXT:    i32 2, label [[LOOP_EXIT]]
21; CHECK-NEXT:    ], !prof [[PROF0:![0-9]+]]
22; CHECK:       loop.latch.peel:
23; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT:%.*]]
24; CHECK:       loop.peel.next:
25; CHECK-NEXT:    br label [[LOOP_PEEL2:%.*]]
26; CHECK:       loop.peel2:
27; CHECK-NEXT:    [[X_PEEL3:%.*]] = call i32 @get.x()
28; CHECK-NEXT:    switch i32 [[X_PEEL3]], label [[LOOP_LATCH_PEEL4:%.*]] [
29; CHECK-NEXT:    i32 0, label [[LOOP_LATCH_PEEL4]]
30; CHECK-NEXT:    i32 1, label [[LOOP_EXIT]]
31; CHECK-NEXT:    i32 2, label [[LOOP_EXIT]]
32; CHECK-NEXT:    ], !prof [[PROF1:![0-9]+]]
33; CHECK:       loop.latch.peel4:
34; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT1:%.*]]
35; CHECK:       loop.peel.next1:
36; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT5:%.*]]
37; CHECK:       loop.peel.next5:
38; CHECK-NEXT:    br label [[ENTRY_PEEL_NEWPH:%.*]]
39; CHECK:       entry.peel.newph:
40; CHECK-NEXT:    br label [[LOOP:%.*]]
41; CHECK:       loop:
42; CHECK-NEXT:    [[X:%.*]] = call i32 @get.x()
43; CHECK-NEXT:    switch i32 [[X]], label [[LOOP_LATCH:%.*]] [
44; CHECK-NEXT:    i32 0, label [[LOOP_LATCH]]
45; CHECK-NEXT:    i32 1, label [[LOOP_EXIT_LOOPEXIT:%.*]]
46; CHECK-NEXT:    i32 2, label [[LOOP_EXIT_LOOPEXIT]]
47; CHECK-NEXT:    ], !prof [[PROF2:![0-9]+]]
48; CHECK:       loop.latch:
49; CHECK-NEXT:    br label [[LOOP]], !llvm.loop [[LOOP3:![0-9]+]]
50; CHECK:       loop.exit.loopexit:
51; CHECK-NEXT:    br label [[LOOP_EXIT]]
52; CHECK:       loop.exit:
53; CHECK-NEXT:    ret void
54
55; DISABLEADV-LABEL: @test()
56; DISABLEADV-NEXT: entry:
57; DISABLEADV-NEXT:  br label %loop
58; DISABLEADV: loop
59; DISABLEADV-NEXT:  %x = call i32 @get.x()
60; DISABLEADV-NEXT:  switch i32 %x, label %loop.latch [
61; DISABLEADV-NEXT:    i32 0, label %loop.latch
62; DISABLEADV-NEXT:    i32 1, label %loop.exit
63; DISABLEADV-NEXT:    i32 2, label %loop.exit
64; DISABLEADV-NEXT:  ], !prof !0
65; DISABLEADV: loop.latch:
66; DISABLEADV-NEXT:  br label %loop
67; DISABLEADV: loop.exit:
68; DISABLEADV-NEXT:  ret void
69
70entry:
71  br label %loop
72
73loop:
74  %x = call i32 @get.x()
75  switch i32 %x, label %loop.latch [
76  i32 0, label %loop.latch
77  i32 1, label %loop.exit
78  i32 2, label %loop.exit
79  ], !prof !0
80
81loop.latch:
82  br label %loop
83
84loop.exit:
85  ret void
86}
87
88!0 = !{!"branch_weights", i32 100, i32 200, i32 20, i32 10}
89
90;.
91; CHECK: [[PROF0]] = !{!"branch_weights", i32 100, i32 200, i32 20, i32 10}
92; CHECK: [[PROF1]] = !{!"branch_weights", i32 90, i32 180, i32 20, i32 10}
93; CHECK: [[PROF2]] = !{!"branch_weights", i32 80, i32 160, i32 20, i32 10}
94; CHECK: [[LOOP3]] = distinct !{!3, !4, !5}
95; CHECK: [[META4:![0-9]+]] = !{!"llvm.loop.peeled.count", i32 2}
96; CHECK: [[META5:![0-9]+]] = !{!"llvm.loop.unroll.disable"}
97;.
98