1 ; RUN: opt < %s -analyze -branch-prob | FileCheck %s
3 define i32 @test1(i32 %i, i32* %a) {
4 ; CHECK: Printing analysis {{.*}} for function 'test1'
7 ; CHECK: edge entry -> body probability is 16 / 16 = 100%
10 %iv = phi i32 [ 0, %entry ], [ %next, %body ]
11 %base = phi i32 [ 0, %entry ], [ %sum, %body ]
12 %arrayidx = getelementptr inbounds i32* %a, i32 %iv
13 %0 = load i32* %arrayidx
14 %sum = add nsw i32 %0, %base
15 %next = add i32 %iv, 1
16 %exitcond = icmp eq i32 %next, %i
17 br i1 %exitcond, label %exit, label %body
18 ; CHECK: edge body -> exit probability is 4 / 128
19 ; CHECK: edge body -> body probability is 124 / 128
25 define i32 @test2(i32 %i, i32 %a, i32 %b) {
26 ; CHECK: Printing analysis {{.*}} for function 'test2'
28 %cond = icmp ult i32 %i, 42
29 br i1 %cond, label %then, label %else, !prof !0
30 ; CHECK: edge entry -> then probability is 64 / 68
31 ; CHECK: edge entry -> else probability is 4 / 68
35 ; CHECK: edge then -> exit probability is 16 / 16 = 100%
39 ; CHECK: edge else -> exit probability is 16 / 16 = 100%
42 %result = phi i32 [ %a, %then ], [ %b, %else ]
46 !0 = metadata !{metadata !"branch_weights", i32 64, i32 4}
48 define i32 @test3(i32 %i, i32 %a, i32 %b, i32 %c, i32 %d, i32 %e) {
49 ; CHECK: Printing analysis {{.*}} for function 'test3'
51 switch i32 %i, label %case_a [ i32 1, label %case_b
54 i32 4, label %case_e ], !prof !1
55 ; CHECK: edge entry -> case_a probability is 4 / 80
56 ; CHECK: edge entry -> case_b probability is 4 / 80
57 ; CHECK: edge entry -> case_c probability is 64 / 80
58 ; CHECK: edge entry -> case_d probability is 4 / 80
59 ; CHECK: edge entry -> case_e probability is 4 / 80
63 ; CHECK: edge case_a -> exit probability is 16 / 16 = 100%
67 ; CHECK: edge case_b -> exit probability is 16 / 16 = 100%
71 ; CHECK: edge case_c -> exit probability is 16 / 16 = 100%
75 ; CHECK: edge case_d -> exit probability is 16 / 16 = 100%
79 ; CHECK: edge case_e -> exit probability is 16 / 16 = 100%
82 %result = phi i32 [ %a, %case_a ],
90 !1 = metadata !{metadata !"branch_weights", i32 4, i32 4, i32 64, i32 4, i32 4}
92 define i32 @test4(i32 %x) nounwind uwtable readnone ssp {
93 ; CHECK: Printing analysis {{.*}} for function 'test4'
95 %conv = sext i32 %x to i64
96 switch i64 %conv, label %return [
102 ; CHECK: edge entry -> return probability is 7 / 85
103 ; CHECK: edge entry -> sw.bb probability is 14 / 85
104 ; CHECK: edge entry -> sw.bb1 probability is 64 / 85
113 %retval.0 = phi i32 [ 5, %sw.bb1 ], [ 1, %sw.bb ], [ 0, %entry ]
117 !2 = metadata !{metadata !"branch_weights", i32 7, i32 6, i32 4, i32 4, i32 64}
119 declare void @coldfunc() cold
121 define i32 @test5(i32 %a, i32 %b, i1 %flag) {
122 ; CHECK: Printing analysis {{.*}} for function 'test5'
124 br i1 %flag, label %then, label %else
125 ; CHECK: edge entry -> then probability is 4 / 68
126 ; CHECK: edge entry -> else probability is 64 / 68
129 call void @coldfunc()
131 ; CHECK: edge then -> exit probability is 16 / 16 = 100%
135 ; CHECK: edge else -> exit probability is 16 / 16 = 100%
138 %result = phi i32 [ %a, %then ], [ %b, %else ]
142 declare i32 @regular_function(i32 %i)
144 define i32 @test_cold_call_sites(i32* %a) {
145 ; Test that edges to blocks post-dominated by cold call sites
146 ; are marked as not expected to be taken.
147 ; TODO(dnovillo) The calls to regular_function should not be merged, but
148 ; they are currently being merged. Convert this into a code generation test
149 ; after that is fixed.
151 ; CHECK: Printing analysis {{.*}} for function 'test_cold_call_sites'
152 ; CHECK: edge entry -> then probability is 4 / 68 = 5.88235%
153 ; CHECK: edge entry -> else probability is 64 / 68 = 94.1176% [HOT edge]
156 %gep1 = getelementptr i32* %a, i32 1
157 %val1 = load i32* %gep1
158 %cond1 = icmp ugt i32 %val1, 1
159 br i1 %cond1, label %then, label %else
162 ; This function is not declared cold, but this call site is.
163 %val4 = call i32 @regular_function(i32 %val1) cold
167 %gep2 = getelementptr i32* %a, i32 2
168 %val2 = load i32* %gep2
169 %val3 = call i32 @regular_function(i32 %val2)
173 %ret = phi i32 [ %val4, %then ], [ %val3, %else ]