Skip to content

Commit 6a137fb

Browse files
[memprof] Use namespaces in a unit test (#119144)
MemProfTest.cpp is about MemProf, so mentioning llvm::memprof everywhere is quite verbose.
1 parent 3733ed6 commit 6a137fb

File tree

1 file changed

+42
-58
lines changed

1 file changed

+42
-58
lines changed

llvm/unittests/ProfileData/MemProfTest.cpp

Lines changed: 42 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@
2222

2323
#include <initializer_list>
2424

25+
namespace llvm {
26+
namespace memprof {
2527
namespace {
2628

2729
using ::llvm::DIGlobal;
@@ -30,22 +32,6 @@ using ::llvm::DILineInfo;
3032
using ::llvm::DILineInfoSpecifier;
3133
using ::llvm::DILocal;
3234
using ::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;
4935
using ::llvm::object::SectionedAddress;
5036
using ::llvm::symbolize::SymbolizableModule;
5137
using ::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

273259
TEST(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

302287
TEST(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() {
553537
TEST(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

577560
TEST(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.
602584
TEST(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.
616598
TEST(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.
638620
TEST(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.
666648
TEST(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

Comments
 (0)