Skip to content

Commit bcc71ab

Browse files
committed
[PGO] Test graceful handling of zero profile counts
This crashed before making fixFuncEntryCount gracefully handle zero counts.
1 parent 320a89e commit bcc71ab

File tree

1 file changed

+42
-0
lines changed

1 file changed

+42
-0
lines changed
Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
; RUN: rm -rf %t && split-file %s %t
2+
3+
; RUN: llvm-profdata merge %t/main.proftext -o %t/main.profdata
4+
; RUN: opt < %t/main.ll -passes=pgo-instr-use -pgo-test-profile-file=%t/main.profdata -S | FileCheck %s
5+
6+
;--- main.ll
7+
8+
; Instrumentation PGO sampling makes corrupt looking counters possible. This
9+
; tests one extreme case:
10+
; Test loading zero profile counts for all instrumented blocks while the entry
11+
; block is not instrumented. Additionally include a non-zero profile count for
12+
; a select instruction, which prevents short circuiting the PGO application.
13+
14+
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
15+
target triple = "x86_64-unknown-linux-gnu"
16+
17+
define i32 @test_no_entry_block_counter(i32 %n) {
18+
; CHECK: define i32 @test_no_entry_block_counter(i32 %n)
19+
; CHECK-SAME: !prof ![[ENTRY_COUNT:[0-9]*]]
20+
entry:
21+
%cmp = icmp slt i32 42, %n
22+
br i1 %cmp, label %tail1, label %tail2
23+
tail1:
24+
%ret = select i1 true, i32 %n, i32 42
25+
; CHECK: %ret = select i1 true, i32 %n, i32 42
26+
; CHECK-SAME: !prof ![[BW_FOR_SELECT:[0-9]+]]
27+
ret i32 %ret
28+
tail2:
29+
ret i32 42
30+
}
31+
; CHECK: ![[ENTRY_COUNT]] = !{!"function_entry_count", i64 1}
32+
; CHECK: ![[BW_FOR_SELECT]] = !{!"branch_weights", i32 1, i32 0}
33+
34+
;--- main.proftext
35+
:ir
36+
test_no_entry_block_counter
37+
431494656217155589
38+
3
39+
0
40+
0
41+
1
42+

0 commit comments

Comments
 (0)