2222
2323#include < initializer_list>
2424
25+ namespace llvm {
26+ namespace memprof {
2527namespace {
2628
2729using ::llvm::DIGlobal;
@@ -30,22 +32,6 @@ using ::llvm::DILineInfo;
3032using ::llvm::DILineInfoSpecifier;
3133using ::llvm::DILocal;
3234using ::llvm::StringRef;
33- using ::llvm::memprof::CallStackId;
34- using ::llvm::memprof::CallStackMap;
35- using ::llvm::memprof::Frame;
36- using ::llvm::memprof::FrameId;
37- using ::llvm::memprof::hashCallStack;
38- using ::llvm::memprof::IndexedAllocationInfo;
39- using ::llvm::memprof::IndexedMemProfData;
40- using ::llvm::memprof::IndexedMemProfRecord;
41- using ::llvm::memprof::MemInfoBlock;
42- using ::llvm::memprof::MemProfReader;
43- using ::llvm::memprof::MemProfRecord;
44- using ::llvm::memprof::MemProfSchema;
45- using ::llvm::memprof::Meta;
46- using ::llvm::memprof::PortableMemInfoBlock;
47- using ::llvm::memprof::RawMemProfReader;
48- using ::llvm::memprof::SegmentEntry;
4935using ::llvm::object::SectionedAddress;
5036using ::llvm::symbolize::SymbolizableModule;
5137using ::testing::ElementsAre;
@@ -251,7 +237,7 @@ TEST(MemProf, PortableWrapper) {
251237 /* dealloc_timestamp=*/ 2000 , /* alloc_cpu=*/ 3 ,
252238 /* dealloc_cpu=*/ 4 , /* Histogram=*/ 0 , /* HistogramSize=*/ 0 );
253239
254- const auto Schema = llvm::memprof:: getFullSchema ();
240+ const auto Schema = getFullSchema ();
255241 PortableMemInfoBlock WriteBlock (Info, Schema);
256242
257243 std::string Buffer;
@@ -271,7 +257,7 @@ TEST(MemProf, PortableWrapper) {
271257}
272258
273259TEST (MemProf, RecordSerializationRoundTripVerion2) {
274- const auto Schema = llvm::memprof:: getFullSchema ();
260+ const auto Schema = getFullSchema ();
275261
276262 MemInfoBlock Info (/* size=*/ 16 , /* access_count=*/ 7 , /* alloc_timestamp=*/ 1000 ,
277263 /* dealloc_timestamp=*/ 2000 , /* alloc_cpu=*/ 3 ,
@@ -290,17 +276,16 @@ TEST(MemProf, RecordSerializationRoundTripVerion2) {
290276
291277 std::string Buffer;
292278 llvm::raw_string_ostream OS (Buffer);
293- Record.serialize (Schema, OS, llvm::memprof:: Version2);
279+ Record.serialize (Schema, OS, Version2);
294280
295281 const IndexedMemProfRecord GotRecord = IndexedMemProfRecord::deserialize (
296- Schema, reinterpret_cast <const unsigned char *>(Buffer.data ()),
297- llvm::memprof::Version2);
282+ Schema, reinterpret_cast <const unsigned char *>(Buffer.data ()), Version2);
298283
299284 EXPECT_EQ (Record, GotRecord);
300285}
301286
302287TEST (MemProf, RecordSerializationRoundTripVersion2HotColdSchema) {
303- const auto Schema = llvm::memprof:: getHotColdSchema ();
288+ const auto Schema = getHotColdSchema ();
304289
305290 MemInfoBlock Info;
306291 Info.AllocCount = 11 ;
@@ -334,22 +319,21 @@ TEST(MemProf, RecordSerializationRoundTripVersion2HotColdSchema) {
334319
335320 // Verify that Schema has propagated all the way to the Info field in each
336321 // IndexedAllocationInfo.
337- ASSERT_THAT (Record.AllocSites , :: SizeIs (2 ));
322+ ASSERT_THAT (Record.AllocSites , SizeIs (2 ));
338323 EXPECT_EQ (Record.AllocSites [0 ].Info .getSchema (), SchemaBitSet);
339324 EXPECT_EQ (Record.AllocSites [1 ].Info .getSchema (), SchemaBitSet);
340325
341326 std::string Buffer;
342327 llvm::raw_string_ostream OS (Buffer);
343- Record.serialize (Schema, OS, llvm::memprof:: Version2);
328+ Record.serialize (Schema, OS, Version2);
344329
345330 const IndexedMemProfRecord GotRecord = IndexedMemProfRecord::deserialize (
346- Schema, reinterpret_cast <const unsigned char *>(Buffer.data ()),
347- llvm::memprof::Version2);
331+ Schema, reinterpret_cast <const unsigned char *>(Buffer.data ()), Version2);
348332
349333 // Verify that Schema comes back correctly after deserialization. Technically,
350334 // the comparison between Record and GotRecord below includes the comparison
351335 // of their Schemas, but we'll verify the Schemas on our own.
352- ASSERT_THAT (GotRecord.AllocSites , :: SizeIs (2 ));
336+ ASSERT_THAT (GotRecord.AllocSites , SizeIs (2 ));
353337 EXPECT_EQ (GotRecord.AllocSites [0 ].Info .getSchema (), SchemaBitSet);
354338 EXPECT_EQ (GotRecord.AllocSites [1 ].Info .getSchema (), SchemaBitSet);
355339
@@ -521,10 +505,10 @@ TEST(MemProf, IndexedMemProfRecordToMemProfRecord) {
521505 IndexedRecord.CallSiteIds .push_back (hashCallStack (CS3));
522506 IndexedRecord.CallSiteIds .push_back (hashCallStack (CS4));
523507
524- llvm::memprof:: FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
508+ FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
525509 MemProfData.Frames );
526- llvm::memprof:: CallStackIdConverter<decltype (MemProfData.CallStacks )>
527- CSIdConv ( MemProfData.CallStacks , FrameIdConv);
510+ CallStackIdConverter<decltype (MemProfData.CallStacks )> CSIdConv (
511+ MemProfData.CallStacks , FrameIdConv);
528512
529513 MemProfRecord Record = IndexedRecord.toMemProfRecord (CSIdConv);
530514
@@ -553,18 +537,17 @@ MemInfoBlock makePartialMIB() {
553537TEST (MemProf, MissingCallStackId) {
554538 // Use a non-existent CallStackId to trigger a mapping error in
555539 // toMemProfRecord.
556- IndexedAllocationInfo AI (0xdeadbeefU , makePartialMIB (),
557- llvm::memprof::getHotColdSchema ());
540+ IndexedAllocationInfo AI (0xdeadbeefU , makePartialMIB (), getHotColdSchema ());
558541
559542 IndexedMemProfRecord IndexedMR;
560543 IndexedMR.AllocSites .push_back (AI);
561544
562545 // Create empty maps.
563546 IndexedMemProfData MemProfData;
564- llvm::memprof:: FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
547+ FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
565548 MemProfData.Frames );
566- llvm::memprof:: CallStackIdConverter<decltype (MemProfData.CallStacks )>
567- CSIdConv ( MemProfData.CallStacks , FrameIdConv);
549+ CallStackIdConverter<decltype (MemProfData.CallStacks )> CSIdConv (
550+ MemProfData.CallStacks , FrameIdConv);
568551
569552 // We are only interested in errors, not the return value.
570553 (void )IndexedMR.toMemProfRecord (CSIdConv);
@@ -575,8 +558,7 @@ TEST(MemProf, MissingCallStackId) {
575558}
576559
577560TEST (MemProf, MissingFrameId) {
578- IndexedAllocationInfo AI (0x222 , makePartialMIB (),
579- llvm::memprof::getHotColdSchema ());
561+ IndexedAllocationInfo AI (0x222 , makePartialMIB (), getHotColdSchema ());
580562
581563 IndexedMemProfRecord IndexedMR;
582564 IndexedMR.AllocSites .push_back (AI);
@@ -585,10 +567,10 @@ TEST(MemProf, MissingFrameId) {
585567 IndexedMemProfData MemProfData;
586568 MemProfData.CallStacks .insert ({0x222 , {2 , 3 }});
587569
588- llvm::memprof:: FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
570+ FrameIdConverter<decltype (MemProfData.Frames )> FrameIdConv (
589571 MemProfData.Frames );
590- llvm::memprof:: CallStackIdConverter<decltype (MemProfData.CallStacks )>
591- CSIdConv ( MemProfData.CallStacks , FrameIdConv);
572+ CallStackIdConverter<decltype (MemProfData.CallStacks )> CSIdConv (
573+ MemProfData.CallStacks , FrameIdConv);
592574
593575 // We are only interested in errors, not the return value.
594576 (void )IndexedMR.toMemProfRecord (CSIdConv);
@@ -600,11 +582,11 @@ TEST(MemProf, MissingFrameId) {
600582
601583// Verify CallStackRadixTreeBuilder can handle empty inputs.
602584TEST (MemProf, RadixTreeBuilderEmpty) {
603- llvm::DenseMap<FrameId, llvm::memprof:: LinearFrameId> MemProfFrameIndexes;
585+ llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes;
604586 llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData;
605- llvm::DenseMap<FrameId, llvm::memprof:: FrameStat> FrameHistogram =
606- llvm::memprof:: computeFrameHistogram<FrameId>(MemProfCallStackData);
607- llvm::memprof:: CallStackRadixTreeBuilder<FrameId> Builder;
587+ llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
588+ computeFrameHistogram<FrameId>(MemProfCallStackData);
589+ CallStackRadixTreeBuilder<FrameId> Builder;
608590 Builder.build (std::move (MemProfCallStackData), &MemProfFrameIndexes,
609591 FrameHistogram);
610592 ASSERT_THAT (Builder.getRadixArray (), testing::IsEmpty ());
@@ -614,14 +596,14 @@ TEST(MemProf, RadixTreeBuilderEmpty) {
614596
615597// Verify CallStackRadixTreeBuilder can handle one trivial call stack.
616598TEST (MemProf, RadixTreeBuilderOne) {
617- llvm::DenseMap<FrameId, llvm::memprof:: LinearFrameId> MemProfFrameIndexes = {
599+ llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes = {
618600 {11 , 1 }, {12 , 2 }, {13 , 3 }};
619601 llvm::SmallVector<FrameId> CS1 = {13 , 12 , 11 };
620602 llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData;
621603 MemProfCallStackData.insert ({hashCallStack (CS1), CS1});
622- llvm::DenseMap<FrameId, llvm::memprof:: FrameStat> FrameHistogram =
623- llvm::memprof:: computeFrameHistogram<FrameId>(MemProfCallStackData);
624- llvm::memprof:: CallStackRadixTreeBuilder<FrameId> Builder;
604+ llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
605+ computeFrameHistogram<FrameId>(MemProfCallStackData);
606+ CallStackRadixTreeBuilder<FrameId> Builder;
625607 Builder.build (std::move (MemProfCallStackData), &MemProfFrameIndexes,
626608 FrameHistogram);
627609 EXPECT_THAT (Builder.getRadixArray (),
@@ -636,16 +618,16 @@ TEST(MemProf, RadixTreeBuilderOne) {
636618
637619// Verify CallStackRadixTreeBuilder can form a link between two call stacks.
638620TEST (MemProf, RadixTreeBuilderTwo) {
639- llvm::DenseMap<FrameId, llvm::memprof:: LinearFrameId> MemProfFrameIndexes = {
621+ llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes = {
640622 {11 , 1 }, {12 , 2 }, {13 , 3 }};
641623 llvm::SmallVector<FrameId> CS1 = {12 , 11 };
642624 llvm::SmallVector<FrameId> CS2 = {13 , 12 , 11 };
643625 llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData;
644626 MemProfCallStackData.insert ({hashCallStack (CS1), CS1});
645627 MemProfCallStackData.insert ({hashCallStack (CS2), CS2});
646- llvm::DenseMap<FrameId, llvm::memprof:: FrameStat> FrameHistogram =
647- llvm::memprof:: computeFrameHistogram<FrameId>(MemProfCallStackData);
648- llvm::memprof:: CallStackRadixTreeBuilder<FrameId> Builder;
628+ llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
629+ computeFrameHistogram<FrameId>(MemProfCallStackData);
630+ CallStackRadixTreeBuilder<FrameId> Builder;
649631 Builder.build (std::move (MemProfCallStackData), &MemProfFrameIndexes,
650632 FrameHistogram);
651633 EXPECT_THAT (Builder.getRadixArray (),
@@ -664,7 +646,7 @@ TEST(MemProf, RadixTreeBuilderTwo) {
664646// Verify CallStackRadixTreeBuilder can form a jump to a prefix that itself has
665647// another jump to another prefix.
666648TEST (MemProf, RadixTreeBuilderSuccessiveJumps) {
667- llvm::DenseMap<FrameId, llvm::memprof:: LinearFrameId> MemProfFrameIndexes = {
649+ llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes = {
668650 {11 , 1 }, {12 , 2 }, {13 , 3 }, {14 , 4 }, {15 , 5 }, {16 , 6 }, {17 , 7 }, {18 , 8 },
669651 };
670652 llvm::SmallVector<FrameId> CS1 = {14 , 13 , 12 , 11 };
@@ -676,9 +658,9 @@ TEST(MemProf, RadixTreeBuilderSuccessiveJumps) {
676658 MemProfCallStackData.insert ({hashCallStack (CS2), CS2});
677659 MemProfCallStackData.insert ({hashCallStack (CS3), CS3});
678660 MemProfCallStackData.insert ({hashCallStack (CS4), CS4});
679- llvm::DenseMap<FrameId, llvm::memprof:: FrameStat> FrameHistogram =
680- llvm::memprof:: computeFrameHistogram<FrameId>(MemProfCallStackData);
681- llvm::memprof:: CallStackRadixTreeBuilder<FrameId> Builder;
661+ llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
662+ computeFrameHistogram<FrameId>(MemProfCallStackData);
663+ CallStackRadixTreeBuilder<FrameId> Builder;
682664 Builder.build (std::move (MemProfCallStackData), &MemProfFrameIndexes,
683665 FrameHistogram);
684666 EXPECT_THAT (Builder.getRadixArray (),
@@ -731,7 +713,7 @@ TEST(MemProf, YAMLParser) {
731713 - {Function: 0x800, LineOffset: 88, Column: 80, IsInlineFrame: false}
732714)YAML" ;
733715
734- llvm::memprof:: YAMLMemProfReader YAMLReader;
716+ YAMLMemProfReader YAMLReader;
735717 YAMLReader.parse (YAMLData);
736718 IndexedMemProfData MemProfData = YAMLReader.takeMemProfData ();
737719
@@ -802,7 +784,7 @@ TEST(MemProf, YAMLWriterMIB) {
802784 MIB.TotalSize = 222 ;
803785 MIB.TotalLifetime = 333 ;
804786 MIB.TotalLifetimeAccessDensity = 444 ;
805- PortableMemInfoBlock PMIB (MIB, llvm::memprof:: getHotColdSchema ());
787+ PortableMemInfoBlock PMIB (MIB, getHotColdSchema ());
806788
807789 std::string Out = serializeInYAML (PMIB);
808790 EXPECT_EQ (Out, R"YAML( ---
@@ -814,3 +796,5 @@ TotalLifetimeAccessDensity: 444
814796)YAML" );
815797}
816798} // namespace
799+ } // namespace memprof
800+ } // namespace llvm
0 commit comments