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(new DominatorTree(F)); 44 LI.reset(new LoopInfo(*DT)); 45 BPI.reset(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(SummaryString, ProfKind, NumCounts, IsPartialProfile, 116 PartialProfileRatio, HotNumCounts, ColdNumCounts) 117 .str(); 118 return parseAssemblyString(formatv(ModuleString, Summary).str(), Err, C); 119 } else 120 return parseAssemblyString(formatv(ModuleString, "").str(), Err, C); 121 } 122 }; 123 124 TEST_F(ProfileSummaryInfoTest, TestNoProfile) { 125 auto M = makeLLVMModule(/*ProfKind=*/nullptr); 126 Function *F = M->getFunction("f"); 127 128 ProfileSummaryInfo PSI = buildPSI(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(0); 144 145 BlockFrequencyInfo BFI = buildBFI(*F); 146 EXPECT_FALSE(PSI.isHotBlock(&BB0, &BFI)); 147 EXPECT_FALSE(PSI.isColdBlock(&BB0, &BFI)); 148 149 CallBase &CS1 = cast<CallBase>(*BB1->getFirstNonPHIIt()); 150 EXPECT_FALSE(PSI.isHotCallSite(CS1, &BFI)); 151 EXPECT_FALSE(PSI.isColdCallSite(CS1, &BFI)); 152 } 153 TEST_F(ProfileSummaryInfoTest, TestCommon) { 154 auto M = makeLLVMModule("InstrProf"); 155 Function *F = M->getFunction("f"); 156 Function *G = M->getFunction("g"); 157 Function *H = M->getFunction("h"); 158 159 ProfileSummaryInfo PSI = buildPSI(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("InstrProf"); 197 Function *F = M->getFunction("f"); 198 ProfileSummaryInfo PSI = buildPSI(M.get()); 199 EXPECT_TRUE(PSI.hasProfileSummary()); 200 EXPECT_TRUE(PSI.hasInstrumentationProfile()); 201 202 BasicBlock &BB0 = F->getEntryBlock(); 203 BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(0); 204 BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(1); 205 BasicBlock *BB3 = BB1->getSingleSuccessor(); 206 207 BlockFrequencyInfo BFI = buildBFI(*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>(*BB1->getFirstNonPHIIt()); 244 BasicBlock::iterator CI2 = BB2->getFirstNonPHIIt(); 245 CallBase &CS2 = cast<CallBase>(*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(llvm::LLVMContext::MD_prof, MDB.createBranchWeights({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("InstrProf"); 264 Function *F = M->getFunction("l"); 265 ProfileSummaryInfo PSI = buildPSI(M.get()); 266 EXPECT_TRUE(PSI.hasProfileSummary()); 267 EXPECT_TRUE(PSI.hasInstrumentationProfile()); 268 269 BasicBlock &BB0 = F->getEntryBlock(); 270 BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(0); 271 BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(1); 272 BasicBlock *BB3 = BB1->getSingleSuccessor(); 273 274 BlockFrequencyInfo BFI = buildBFI(*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("SampleProfile"); 292 Function *F = M->getFunction("f"); 293 ProfileSummaryInfo PSI = buildPSI(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(0); 300 BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(1); 301 BasicBlock *BB3 = BB1->getSingleSuccessor(); 302 303 BlockFrequencyInfo BFI = buildBFI(*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>(*BB1->getFirstNonPHIIt()); 340 BasicBlock::iterator CI2 = BB2->getFirstNonPHIIt(); 341 // Manually attach branch weights metadata to the call instruction. 342 SmallVector<uint32_t, 1> Weights; 343 Weights.push_back(1000); 344 MDBuilder MDB(M->getContext()); 345 CI2->setMetadata(LLVMContext::MD_prof, MDB.createBranchWeights(Weights)); 346 CallBase &CS2 = cast<CallBase>(*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(llvm::LLVMContext::MD_prof, MDB.createBranchWeights({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("SampleProfile"); 364 Function *F = M->getFunction("l"); 365 ProfileSummaryInfo PSI = buildPSI(M.get()); 366 EXPECT_TRUE(PSI.hasProfileSummary()); 367 EXPECT_TRUE(PSI.hasSampleProfile()); 368 369 BasicBlock &BB0 = F->getEntryBlock(); 370 BasicBlock *BB1 = BB0.getTerminator()->getSuccessor(0); 371 BasicBlock *BB2 = BB0.getTerminator()->getSuccessor(1); 372 BasicBlock *BB3 = BB1->getSingleSuccessor(); 373 374 BlockFrequencyInfo BFI = buildBFI(*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(true); 392 393 // With PartialProfileRatio unset (zero.) 394 auto M1 = makeLLVMModule("SampleProfile", /*NumCounts*/ 3, 395 /*IsPartialProfile*/ 1, 396 /*PartialProfileRatio*/ 0.0, 397 /*HotNumCounts*/ 3, /*ColdNumCounts*/ 10); 398 ProfileSummaryInfo PSI1 = buildPSI(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("SampleProfile", /*NumCounts*/ 27493235, 407 /*IsPartialProfile*/ 1, 408 /*PartialProfileRatio*/ 0.00000012, 409 /*HotNumCounts*/ 3102082, 410 /*ColdNumCounts*/ 18306149); 411 ProfileSummaryInfo PSI2 = buildPSI(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("SampleProfile", /*NumCounts*/ 27493235, 420 /*IsPartialProfile*/ 1, 421 /*PartialProfileRatio*/ 0.9, 422 /*HotNumCounts*/ 3102082, 423 /*ColdNumCounts*/ 18306149); 424 ProfileSummaryInfo PSI3 = buildPSI(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