@@ -581,11 +581,11 @@ TEST(MemProf, MissingFrameId) {
581
581
// Verify CallStackRadixTreeBuilder can handle empty inputs.
582
582
TEST (MemProf, RadixTreeBuilderEmpty) {
583
583
llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes;
584
- llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData ;
584
+ IndexedMemProfData MemProfData ;
585
585
llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
586
- computeFrameHistogram<FrameId>(MemProfCallStackData );
586
+ computeFrameHistogram<FrameId>(MemProfData. CallStacks );
587
587
CallStackRadixTreeBuilder<FrameId> Builder;
588
- Builder.build (std::move (MemProfCallStackData ), &MemProfFrameIndexes,
588
+ Builder.build (std::move (MemProfData. CallStacks ), &MemProfFrameIndexes,
589
589
FrameHistogram);
590
590
ASSERT_THAT (Builder.getRadixArray (), IsEmpty ());
591
591
const auto Mappings = Builder.takeCallStackPos ();
@@ -597,12 +597,12 @@ TEST(MemProf, RadixTreeBuilderOne) {
597
597
llvm::DenseMap<FrameId, LinearFrameId> MemProfFrameIndexes = {
598
598
{11 , 1 }, {12 , 2 }, {13 , 3 }};
599
599
llvm::SmallVector<FrameId> CS1 = {13 , 12 , 11 };
600
- llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData ;
601
- MemProfCallStackData. insert ({ hashCallStack ( CS1), CS1} );
600
+ IndexedMemProfData MemProfData ;
601
+ MemProfData. addCallStack ( CS1);
602
602
llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
603
- computeFrameHistogram<FrameId>(MemProfCallStackData );
603
+ computeFrameHistogram<FrameId>(MemProfData. CallStacks );
604
604
CallStackRadixTreeBuilder<FrameId> Builder;
605
- Builder.build (std::move (MemProfCallStackData ), &MemProfFrameIndexes,
605
+ Builder.build (std::move (MemProfData. CallStacks ), &MemProfFrameIndexes,
606
606
FrameHistogram);
607
607
EXPECT_THAT (Builder.getRadixArray (),
608
608
ElementsAre (3U , // Size of CS1,
@@ -620,13 +620,13 @@ TEST(MemProf, RadixTreeBuilderTwo) {
620
620
{11 , 1 }, {12 , 2 }, {13 , 3 }};
621
621
llvm::SmallVector<FrameId> CS1 = {12 , 11 };
622
622
llvm::SmallVector<FrameId> CS2 = {13 , 12 , 11 };
623
- llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData ;
624
- MemProfCallStackData. insert ({ hashCallStack ( CS1), CS1} );
625
- MemProfCallStackData. insert ({ hashCallStack ( CS2), CS2} );
623
+ IndexedMemProfData MemProfData ;
624
+ MemProfData. addCallStack ( CS1);
625
+ MemProfData. addCallStack ( CS2);
626
626
llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
627
- computeFrameHistogram<FrameId>(MemProfCallStackData );
627
+ computeFrameHistogram<FrameId>(MemProfData. CallStacks );
628
628
CallStackRadixTreeBuilder<FrameId> Builder;
629
- Builder.build (std::move (MemProfCallStackData ), &MemProfFrameIndexes,
629
+ Builder.build (std::move (MemProfData. CallStacks ), &MemProfFrameIndexes,
630
630
FrameHistogram);
631
631
EXPECT_THAT (Builder.getRadixArray (),
632
632
ElementsAre (2U , // Size of CS1
@@ -651,15 +651,15 @@ TEST(MemProf, RadixTreeBuilderSuccessiveJumps) {
651
651
llvm::SmallVector<FrameId> CS2 = {15 , 13 , 12 , 11 };
652
652
llvm::SmallVector<FrameId> CS3 = {17 , 16 , 12 , 11 };
653
653
llvm::SmallVector<FrameId> CS4 = {18 , 16 , 12 , 11 };
654
- llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData ;
655
- MemProfCallStackData. insert ({ hashCallStack ( CS1), CS1} );
656
- MemProfCallStackData. insert ({ hashCallStack ( CS2), CS2} );
657
- MemProfCallStackData. insert ({ hashCallStack ( CS3), CS3} );
658
- MemProfCallStackData. insert ({ hashCallStack ( CS4), CS4} );
654
+ IndexedMemProfData MemProfData ;
655
+ MemProfData. addCallStack ( CS1);
656
+ MemProfData. addCallStack ( CS2);
657
+ MemProfData. addCallStack ( CS3);
658
+ MemProfData. addCallStack ( CS4);
659
659
llvm::DenseMap<FrameId, FrameStat> FrameHistogram =
660
- computeFrameHistogram<FrameId>(MemProfCallStackData );
660
+ computeFrameHistogram<FrameId>(MemProfData. CallStacks );
661
661
CallStackRadixTreeBuilder<FrameId> Builder;
662
- Builder.build (std::move (MemProfCallStackData ), &MemProfFrameIndexes,
662
+ Builder.build (std::move (MemProfData. CallStacks ), &MemProfFrameIndexes,
663
663
FrameHistogram);
664
664
EXPECT_THAT (Builder.getRadixArray (),
665
665
ElementsAre (4U , // Size of CS1
0 commit comments