1 | //===- ProfileSummaryInfoTest.cpp - ProfileSummaryInfo unit tests ---------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "llvm/Analysis/ProfileSummaryInfo.h" |
10 | #include "llvm/Analysis/BlockFrequencyInfo.h" |
11 | #include "llvm/Analysis/BranchProbabilityInfo.h" |
12 | #include "llvm/Analysis/LoopInfo.h" |
13 | #include "llvm/AsmParser/Parser.h" |
14 | #include "llvm/IR/BasicBlock.h" |
15 | #include "llvm/IR/Dominators.h" |
16 | #include "llvm/IR/Function.h" |
17 | #include "llvm/IR/LLVMContext.h" |
18 | #include "llvm/IR/MDBuilder.h" |
19 | #include "llvm/IR/Module.h" |
20 | #include "llvm/Support/CommandLine.h" |
21 | #include "llvm/Support/DataTypes.h" |
22 | #include "llvm/Support/FormatVariadic.h" |
23 | #include "llvm/Support/SourceMgr.h" |
24 | #include "llvm/Support/raw_ostream.h" |
25 | #include "gtest/gtest.h" |
26 | |
27 | extern llvm::cl::opt<bool> ScalePartialSampleProfileWorkingSetSize; |
28 | |
29 | namespace llvm { |
30 | namespace { |
31 | |
32 | class ProfileSummaryInfoTest : public testing::Test { |
33 | protected: |
34 | LLVMContext C; |
35 | std::unique_ptr<BranchProbabilityInfo> BPI; |
36 | std::unique_ptr<DominatorTree> DT; |
37 | std::unique_ptr<LoopInfo> LI; |
38 | |
39 | ProfileSummaryInfo buildPSI(Module *M) { |
40 | return ProfileSummaryInfo(*M); |
41 | } |
42 | BlockFrequencyInfo buildBFI(Function &F) { |
43 | DT.reset(p: new DominatorTree(F)); |
44 | LI.reset(p: new LoopInfo(*DT)); |
45 | BPI.reset(p: new BranchProbabilityInfo(F, *LI)); |
46 | return BlockFrequencyInfo(F, *BPI, *LI); |
47 | } |
48 | std::unique_ptr<Module> makeLLVMModule(const char *ProfKind = nullptr, |
49 | uint64_t NumCounts = 3, |
50 | uint64_t IsPartialProfile = 0, |
51 | double PartialProfileRatio = 0.0, |
52 | uint64_t HotNumCounts = 3, |
53 | uint64_t ColdNumCounts = 10) { |
54 | const char *ModuleString = |
55 | "define i32 @g(i32 %x) !prof !21 {{\n" |
56 | " ret i32 0\n" |
57 | "}\n" |
58 | "define i32 @h(i32 %x) !prof !22 {{\n" |
59 | " ret i32 0\n" |
60 | "}\n" |
61 | "define i32 @f(i32 %x) !prof !20 {{\n" |
62 | "bb0:\n" |
63 | " %y1 = icmp eq i32 %x, 0 \n" |
64 | " br i1 %y1, label %bb1, label %bb2, !prof !23 \n" |
65 | "bb1:\n" |
66 | " %z1 = call i32 @g(i32 %x)\n" |
67 | " br label %bb3\n" |
68 | "bb2:\n" |
69 | " %z2 = call i32 @h(i32 %x)\n" |
70 | " br label %bb3\n" |
71 | "bb3:\n" |
72 | " %y2 = phi i32 [0, %bb1], [1, %bb2] \n" |
73 | " ret i32 %y2\n" |
74 | "}\n" |
75 | "define i32 @l(i32 %x) {{\n" |
76 | "bb0:\n" |
77 | " %y1 = icmp eq i32 %x, 0 \n" |
78 | " br i1 %y1, label %bb1, label %bb2, !prof !23 \n" |
79 | "bb1:\n" |
80 | " %z1 = call i32 @g(i32 %x)\n" |
81 | " br label %bb3\n" |
82 | "bb2:\n" |
83 | " %z2 = call i32 @h(i32 %x)\n" |
84 | " br label %bb3\n" |
85 | "bb3:\n" |
86 | " %y2 = phi i32 [0, %bb1], [1, %bb2] \n" |
87 | " ret i32 %y2\n" |
88 | "}\n" |
89 | "!20 = !{{!\"function_entry_count\", i64 400}\n" |
90 | "!21 = !{{!\"function_entry_count\", i64 1}\n" |
91 | "!22 = !{{!\"function_entry_count\", i64 100}\n" |
92 | "!23 = !{{!\"branch_weights\", i32 64, i32 4}\n" |
93 | "{0}" ; |
94 | const char *SummaryString = |
95 | "!llvm.module.flags = !{{!1}\n" |
96 | "!1 = !{{i32 1, !\"ProfileSummary\", !2}\n" |
97 | "!2 = !{{!3, !4, !5, !6, !7, !8, !9, !10, !11, !12}\n" |
98 | "!3 = !{{!\"ProfileFormat\", !\"{0}\"}\n" |
99 | "!4 = !{{!\"TotalCount\", i64 10000}\n" |
100 | "!5 = !{{!\"MaxCount\", i64 10}\n" |
101 | "!6 = !{{!\"MaxInternalCount\", i64 1}\n" |
102 | "!7 = !{{!\"MaxFunctionCount\", i64 1000}\n" |
103 | "!8 = !{{!\"NumCounts\", i64 {1}}\n" |
104 | "!9 = !{{!\"NumFunctions\", i64 3}\n" |
105 | "!10 = !{{!\"IsPartialProfile\", i64 {2}}\n" |
106 | "!11 = !{{!\"PartialProfileRatio\", double {3}}\n" |
107 | "!12 = !{{!\"DetailedSummary\", !13}\n" |
108 | "!13 = !{{!14, !15, !16}\n" |
109 | "!14 = !{{i32 10000, i64 1000, i32 1}\n" |
110 | "!15 = !{{i32 990000, i64 300, i32 {4}}\n" |
111 | "!16 = !{{i32 999999, i64 5, i32 {5}}\n" ; |
112 | SMDiagnostic Err; |
113 | if (ProfKind) { |
114 | auto Summary = |
115 | formatv(Fmt: SummaryString, Vals&: ProfKind, Vals&: NumCounts, Vals&: IsPartialProfile, |
116 | Vals&: PartialProfileRatio, Vals&: HotNumCounts, Vals&: ColdNumCounts) |
117 | .str(); |
118 | return parseAssemblyString(AsmString: formatv(Fmt: ModuleString, Vals&: Summary).str(), Err, Context&: C); |
119 | } else |
120 | return parseAssemblyString(AsmString: formatv(Fmt: ModuleString, Vals: "" ).str(), Err, Context&: C); |
121 | } |
122 | }; |
123 | |
124 | TEST_F(ProfileSummaryInfoTest, TestNoProfile) { |
125 | auto M = makeLLVMModule(/*ProfKind=*/nullptr); |
126 | Function *F = M->getFunction(Name: "f" ); |
127 | |
128 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
129 | EXPECT_FALSE(PSI.hasProfileSummary()); |
130 | EXPECT_FALSE(PSI.hasSampleProfile()); |
131 | EXPECT_FALSE(PSI.hasInstrumentationProfile()); |
132 | // In the absence of profiles, is{Hot|Cold}X methods should always return |
133 | // false. |
134 | EXPECT_FALSE(PSI.isHotCount(1000)); |
135 | EXPECT_FALSE(PSI.isHotCount(0)); |
136 | EXPECT_FALSE(PSI.isColdCount(1000)); |
137 | EXPECT_FALSE(PSI.isColdCount(0)); |
138 | |
139 | EXPECT_FALSE(PSI.isFunctionEntryHot(F)); |
140 | EXPECT_FALSE(PSI.isFunctionEntryCold(F)); |
141 | |
142 | BasicBlock &BB0 = F->getEntryBlock(); |
143 | BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(Idx: 0); |
144 | |
145 | BlockFrequencyInfo BFI = buildBFI(F&: *F); |
146 | EXPECT_FALSE(PSI.isHotBlock(&BB0, &BFI)); |
147 | EXPECT_FALSE(PSI.isColdBlock(&BB0, &BFI)); |
148 | |
149 | CallBase &CS1 = cast<CallBase>(Val&: *BB1->getFirstNonPHI()); |
150 | EXPECT_FALSE(PSI.isHotCallSite(CS1, &BFI)); |
151 | EXPECT_FALSE(PSI.isColdCallSite(CS1, &BFI)); |
152 | } |
153 | TEST_F(ProfileSummaryInfoTest, TestCommon) { |
154 | auto M = makeLLVMModule(ProfKind: "InstrProf" ); |
155 | Function *F = M->getFunction(Name: "f" ); |
156 | Function *G = M->getFunction(Name: "g" ); |
157 | Function *H = M->getFunction(Name: "h" ); |
158 | |
159 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
160 | EXPECT_TRUE(PSI.hasProfileSummary()); |
161 | EXPECT_TRUE(PSI.isHotCount(400)); |
162 | EXPECT_TRUE(PSI.isColdCount(2)); |
163 | EXPECT_FALSE(PSI.isColdCount(100)); |
164 | EXPECT_FALSE(PSI.isHotCount(100)); |
165 | |
166 | EXPECT_TRUE(PSI.isHotCountNthPercentile(990000, 400)); |
167 | EXPECT_FALSE(PSI.isHotCountNthPercentile(990000, 100)); |
168 | EXPECT_FALSE(PSI.isHotCountNthPercentile(990000, 2)); |
169 | |
170 | EXPECT_FALSE(PSI.isColdCountNthPercentile(990000, 400)); |
171 | EXPECT_TRUE(PSI.isColdCountNthPercentile(990000, 100)); |
172 | EXPECT_TRUE(PSI.isColdCountNthPercentile(990000, 2)); |
173 | |
174 | EXPECT_TRUE(PSI.isHotCountNthPercentile(999999, 400)); |
175 | EXPECT_TRUE(PSI.isHotCountNthPercentile(999999, 100)); |
176 | EXPECT_FALSE(PSI.isHotCountNthPercentile(999999, 2)); |
177 | |
178 | EXPECT_FALSE(PSI.isColdCountNthPercentile(999999, 400)); |
179 | EXPECT_FALSE(PSI.isColdCountNthPercentile(999999, 100)); |
180 | EXPECT_TRUE(PSI.isColdCountNthPercentile(999999, 2)); |
181 | |
182 | EXPECT_FALSE(PSI.isHotCountNthPercentile(10000, 400)); |
183 | EXPECT_FALSE(PSI.isHotCountNthPercentile(10000, 100)); |
184 | EXPECT_FALSE(PSI.isHotCountNthPercentile(10000, 2)); |
185 | |
186 | EXPECT_TRUE(PSI.isColdCountNthPercentile(10000, 400)); |
187 | EXPECT_TRUE(PSI.isColdCountNthPercentile(10000, 100)); |
188 | EXPECT_TRUE(PSI.isColdCountNthPercentile(10000, 2)); |
189 | |
190 | EXPECT_TRUE(PSI.isFunctionEntryHot(F)); |
191 | EXPECT_FALSE(PSI.isFunctionEntryHot(G)); |
192 | EXPECT_FALSE(PSI.isFunctionEntryHot(H)); |
193 | } |
194 | |
195 | TEST_F(ProfileSummaryInfoTest, InstrProf) { |
196 | auto M = makeLLVMModule(ProfKind: "InstrProf" ); |
197 | Function *F = M->getFunction(Name: "f" ); |
198 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
199 | EXPECT_TRUE(PSI.hasProfileSummary()); |
200 | EXPECT_TRUE(PSI.hasInstrumentationProfile()); |
201 | |
202 | BasicBlock &BB0 = F->getEntryBlock(); |
203 | BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(Idx: 0); |
204 | BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(Idx: 1); |
205 | BasicBlock *BB3 = BB1->getSingleSuccessor(); |
206 | |
207 | BlockFrequencyInfo BFI = buildBFI(F&: *F); |
208 | EXPECT_TRUE(PSI.isHotBlock(&BB0, &BFI)); |
209 | EXPECT_TRUE(PSI.isHotBlock(BB1, &BFI)); |
210 | EXPECT_FALSE(PSI.isHotBlock(BB2, &BFI)); |
211 | EXPECT_TRUE(PSI.isHotBlock(BB3, &BFI)); |
212 | |
213 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, &BB0, &BFI)); |
214 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, BB1, &BFI)); |
215 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB2, &BFI)); |
216 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, BB3, &BFI)); |
217 | |
218 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, &BB0, &BFI)); |
219 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB1, &BFI)); |
220 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(990000, BB2, &BFI)); |
221 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB3, &BFI)); |
222 | |
223 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, &BB0, &BFI)); |
224 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB1, &BFI)); |
225 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB2, &BFI)); |
226 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB3, &BFI)); |
227 | |
228 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, &BB0, &BFI)); |
229 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB1, &BFI)); |
230 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB2, &BFI)); |
231 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB3, &BFI)); |
232 | |
233 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, &BB0, &BFI)); |
234 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB1, &BFI)); |
235 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB2, &BFI)); |
236 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB3, &BFI)); |
237 | |
238 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, &BB0, &BFI)); |
239 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB1, &BFI)); |
240 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB2, &BFI)); |
241 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB3, &BFI)); |
242 | |
243 | CallBase &CS1 = cast<CallBase>(Val&: *BB1->getFirstNonPHI()); |
244 | auto *CI2 = BB2->getFirstNonPHI(); |
245 | CallBase &CS2 = cast<CallBase>(Val&: *CI2); |
246 | |
247 | EXPECT_TRUE(PSI.isHotCallSite(CS1, &BFI)); |
248 | EXPECT_FALSE(PSI.isHotCallSite(CS2, &BFI)); |
249 | |
250 | // Test that adding an MD_prof metadata with a hot count on CS2 does not |
251 | // change its hotness as it has no effect in instrumented profiling. |
252 | MDBuilder MDB(M->getContext()); |
253 | CI2->setMetadata(KindID: llvm::LLVMContext::MD_prof, Node: MDB.createBranchWeights(Weights: {400})); |
254 | EXPECT_FALSE(PSI.isHotCallSite(CS2, &BFI)); |
255 | |
256 | EXPECT_TRUE(PSI.isFunctionHotInCallGraphNthPercentile(990000, F, BFI)); |
257 | EXPECT_FALSE(PSI.isFunctionColdInCallGraphNthPercentile(990000, F, BFI)); |
258 | EXPECT_FALSE(PSI.isFunctionHotInCallGraphNthPercentile(10000, F, BFI)); |
259 | EXPECT_TRUE(PSI.isFunctionColdInCallGraphNthPercentile(10000, F, BFI)); |
260 | } |
261 | |
262 | TEST_F(ProfileSummaryInfoTest, InstrProfNoFuncEntryCount) { |
263 | auto M = makeLLVMModule(ProfKind: "InstrProf" ); |
264 | Function *F = M->getFunction(Name: "l" ); |
265 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
266 | EXPECT_TRUE(PSI.hasProfileSummary()); |
267 | EXPECT_TRUE(PSI.hasInstrumentationProfile()); |
268 | |
269 | BasicBlock &BB0 = F->getEntryBlock(); |
270 | BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(Idx: 0); |
271 | BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(Idx: 1); |
272 | BasicBlock *BB3 = BB1->getSingleSuccessor(); |
273 | |
274 | BlockFrequencyInfo BFI = buildBFI(F&: *F); |
275 | |
276 | // Without the entry count, all should return false. |
277 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, &BB0, &BFI)); |
278 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB1, &BFI)); |
279 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB2, &BFI)); |
280 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB3, &BFI)); |
281 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, &BB0, &BFI)); |
282 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB1, &BFI)); |
283 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB2, &BFI)); |
284 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB3, &BFI)); |
285 | |
286 | EXPECT_FALSE(PSI.isFunctionHotInCallGraphNthPercentile(990000, F, BFI)); |
287 | EXPECT_FALSE(PSI.isFunctionColdInCallGraphNthPercentile(990000, F, BFI)); |
288 | } |
289 | |
290 | TEST_F(ProfileSummaryInfoTest, SampleProf) { |
291 | auto M = makeLLVMModule(ProfKind: "SampleProfile" ); |
292 | Function *F = M->getFunction(Name: "f" ); |
293 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
294 | EXPECT_TRUE(PSI.hasProfileSummary()); |
295 | EXPECT_TRUE(PSI.hasSampleProfile()); |
296 | EXPECT_FALSE(PSI.hasPartialSampleProfile()); |
297 | |
298 | BasicBlock &BB0 = F->getEntryBlock(); |
299 | BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(Idx: 0); |
300 | BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(Idx: 1); |
301 | BasicBlock *BB3 = BB1->getSingleSuccessor(); |
302 | |
303 | BlockFrequencyInfo BFI = buildBFI(F&: *F); |
304 | EXPECT_TRUE(PSI.isHotBlock(&BB0, &BFI)); |
305 | EXPECT_TRUE(PSI.isHotBlock(BB1, &BFI)); |
306 | EXPECT_FALSE(PSI.isHotBlock(BB2, &BFI)); |
307 | EXPECT_TRUE(PSI.isHotBlock(BB3, &BFI)); |
308 | |
309 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, &BB0, &BFI)); |
310 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, BB1, &BFI)); |
311 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB2, &BFI)); |
312 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(990000, BB3, &BFI)); |
313 | |
314 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, &BB0, &BFI)); |
315 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB1, &BFI)); |
316 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(990000, BB2, &BFI)); |
317 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB3, &BFI)); |
318 | |
319 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, &BB0, &BFI)); |
320 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB1, &BFI)); |
321 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB2, &BFI)); |
322 | EXPECT_TRUE(PSI.isHotBlockNthPercentile(999900, BB3, &BFI)); |
323 | |
324 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, &BB0, &BFI)); |
325 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB1, &BFI)); |
326 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB2, &BFI)); |
327 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(999900, BB3, &BFI)); |
328 | |
329 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, &BB0, &BFI)); |
330 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB1, &BFI)); |
331 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB2, &BFI)); |
332 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(10000, BB3, &BFI)); |
333 | |
334 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, &BB0, &BFI)); |
335 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB1, &BFI)); |
336 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB2, &BFI)); |
337 | EXPECT_TRUE(PSI.isColdBlockNthPercentile(10000, BB3, &BFI)); |
338 | |
339 | CallBase &CS1 = cast<CallBase>(Val&: *BB1->getFirstNonPHI()); |
340 | auto *CI2 = BB2->getFirstNonPHI(); |
341 | // Manually attach branch weights metadata to the call instruction. |
342 | SmallVector<uint32_t, 1> Weights; |
343 | Weights.push_back(Elt: 1000); |
344 | MDBuilder MDB(M->getContext()); |
345 | CI2->setMetadata(KindID: LLVMContext::MD_prof, Node: MDB.createBranchWeights(Weights)); |
346 | CallBase &CS2 = cast<CallBase>(Val&: *CI2); |
347 | |
348 | EXPECT_FALSE(PSI.isHotCallSite(CS1, &BFI)); |
349 | EXPECT_TRUE(PSI.isHotCallSite(CS2, &BFI)); |
350 | |
351 | // Test that CS2 is considered hot when it gets an MD_prof metadata with |
352 | // weights that exceed the hot count threshold. |
353 | CI2->setMetadata(KindID: llvm::LLVMContext::MD_prof, Node: MDB.createBranchWeights(Weights: {400})); |
354 | EXPECT_TRUE(PSI.isHotCallSite(CS2, &BFI)); |
355 | |
356 | EXPECT_TRUE(PSI.isFunctionHotInCallGraphNthPercentile(990000, F, BFI)); |
357 | EXPECT_FALSE(PSI.isFunctionColdInCallGraphNthPercentile(990000, F, BFI)); |
358 | EXPECT_FALSE(PSI.isFunctionHotInCallGraphNthPercentile(10000, F, BFI)); |
359 | EXPECT_TRUE(PSI.isFunctionColdInCallGraphNthPercentile(10000, F, BFI)); |
360 | } |
361 | |
362 | TEST_F(ProfileSummaryInfoTest, SampleProfNoFuncEntryCount) { |
363 | auto M = makeLLVMModule(ProfKind: "SampleProfile" ); |
364 | Function *F = M->getFunction(Name: "l" ); |
365 | ProfileSummaryInfo PSI = buildPSI(M: M.get()); |
366 | EXPECT_TRUE(PSI.hasProfileSummary()); |
367 | EXPECT_TRUE(PSI.hasSampleProfile()); |
368 | |
369 | BasicBlock &BB0 = F->getEntryBlock(); |
370 | BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(Idx: 0); |
371 | BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(Idx: 1); |
372 | BasicBlock *BB3 = BB1->getSingleSuccessor(); |
373 | |
374 | BlockFrequencyInfo BFI = buildBFI(F&: *F); |
375 | |
376 | // Without the entry count, all should return false. |
377 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, &BB0, &BFI)); |
378 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB1, &BFI)); |
379 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB2, &BFI)); |
380 | EXPECT_FALSE(PSI.isHotBlockNthPercentile(990000, BB3, &BFI)); |
381 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, &BB0, &BFI)); |
382 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB1, &BFI)); |
383 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB2, &BFI)); |
384 | EXPECT_FALSE(PSI.isColdBlockNthPercentile(990000, BB3, &BFI)); |
385 | |
386 | EXPECT_FALSE(PSI.isFunctionHotInCallGraphNthPercentile(990000, F, BFI)); |
387 | EXPECT_FALSE(PSI.isFunctionColdInCallGraphNthPercentile(990000, F, BFI)); |
388 | } |
389 | |
390 | TEST_F(ProfileSummaryInfoTest, PartialSampleProfWorkingSetSize) { |
391 | ScalePartialSampleProfileWorkingSetSize.setValue(V: true); |
392 | |
393 | // With PartialProfileRatio unset (zero.) |
394 | auto M1 = makeLLVMModule(ProfKind: "SampleProfile" , /*NumCounts*/ 3, |
395 | /*IsPartialProfile*/ 1, |
396 | /*PartialProfileRatio*/ 0.0, |
397 | /*HotNumCounts*/ 3, /*ColdNumCounts*/ 10); |
398 | ProfileSummaryInfo PSI1 = buildPSI(M: M1.get()); |
399 | EXPECT_TRUE(PSI1.hasProfileSummary()); |
400 | EXPECT_TRUE(PSI1.hasSampleProfile()); |
401 | EXPECT_TRUE(PSI1.hasPartialSampleProfile()); |
402 | EXPECT_FALSE(PSI1.hasHugeWorkingSetSize()); |
403 | EXPECT_FALSE(PSI1.hasLargeWorkingSetSize()); |
404 | |
405 | // With PartialProfileRatio set (non-zero) and a small working set size. |
406 | auto M2 = makeLLVMModule(ProfKind: "SampleProfile" , /*NumCounts*/ 27493235, |
407 | /*IsPartialProfile*/ 1, |
408 | /*PartialProfileRatio*/ 0.00000012, |
409 | /*HotNumCounts*/ 3102082, |
410 | /*ColdNumCounts*/ 18306149); |
411 | ProfileSummaryInfo PSI2 = buildPSI(M: M2.get()); |
412 | EXPECT_TRUE(PSI2.hasProfileSummary()); |
413 | EXPECT_TRUE(PSI2.hasSampleProfile()); |
414 | EXPECT_TRUE(PSI2.hasPartialSampleProfile()); |
415 | EXPECT_FALSE(PSI2.hasHugeWorkingSetSize()); |
416 | EXPECT_FALSE(PSI2.hasLargeWorkingSetSize()); |
417 | |
418 | // With PartialProfileRatio is set (non-zero) and a large working set size. |
419 | auto M3 = makeLLVMModule(ProfKind: "SampleProfile" , /*NumCounts*/ 27493235, |
420 | /*IsPartialProfile*/ 1, |
421 | /*PartialProfileRatio*/ 0.9, |
422 | /*HotNumCounts*/ 3102082, |
423 | /*ColdNumCounts*/ 18306149); |
424 | ProfileSummaryInfo PSI3 = buildPSI(M: M3.get()); |
425 | EXPECT_TRUE(PSI3.hasProfileSummary()); |
426 | EXPECT_TRUE(PSI3.hasSampleProfile()); |
427 | EXPECT_TRUE(PSI3.hasPartialSampleProfile()); |
428 | EXPECT_TRUE(PSI3.hasHugeWorkingSetSize()); |
429 | EXPECT_TRUE(PSI3.hasLargeWorkingSetSize()); |
430 | } |
431 | |
432 | } // end anonymous namespace |
433 | } // end namespace llvm |
434 | |