aboutsummaryrefslogtreecommitdiff
path: root/llvm/unittests/ProfileData/MemProfTest.cpp
diff options
context:
space:
mode:
authorTeresa Johnson <tejohnson@google.com>2024-11-20 10:08:58 -0800
committerGitHub <noreply@github.com>2024-11-20 10:08:58 -0800
commite14827f0828d14ef17ab76316e8449d1b76e2617 (patch)
treecdc599c1cb9f81ae0b58497367f109ff5c1e1ac8 /llvm/unittests/ProfileData/MemProfTest.cpp
parent6473a36edc571cf0734a2e8d4354e332efb170e9 (diff)
downloadllvm-e14827f0828d14ef17ab76316e8449d1b76e2617.zip
llvm-e14827f0828d14ef17ab76316e8449d1b76e2617.tar.gz
llvm-e14827f0828d14ef17ab76316e8449d1b76e2617.tar.bz2
[MemProf] Templatize CallStackRadixTreeBuilder (NFC) (#117014)
Prepare for usage in the bitcode reader/writer where we already have a LinearFrameId: - templatize input frame id type in CallStackRadixTreeBuilder - templatize input frame id type in computeFrameHistogram - make the map from FrameId to LinearFrameId optional We plan to use the same radix format in the ThinLTO summary records, where we already have a LinearFrameId.
Diffstat (limited to 'llvm/unittests/ProfileData/MemProfTest.cpp')
-rw-r--r--llvm/unittests/ProfileData/MemProfTest.cpp16
1 files changed, 8 insertions, 8 deletions
diff --git a/llvm/unittests/ProfileData/MemProfTest.cpp b/llvm/unittests/ProfileData/MemProfTest.cpp
index e5d41db..e68cc09 100644
--- a/llvm/unittests/ProfileData/MemProfTest.cpp
+++ b/llvm/unittests/ProfileData/MemProfTest.cpp
@@ -659,8 +659,8 @@ TEST(MemProf, RadixTreeBuilderEmpty) {
llvm::MapVector<CallStackId, llvm::SmallVector<FrameId>> MemProfCallStackData;
llvm::DenseMap<llvm::memprof::FrameId, llvm::memprof::FrameStat>
FrameHistogram =
- llvm::memprof::computeFrameHistogram(MemProfCallStackData);
- llvm::memprof::CallStackRadixTreeBuilder Builder;
+ llvm::memprof::computeFrameHistogram<FrameId>(MemProfCallStackData);
+ llvm::memprof::CallStackRadixTreeBuilder<FrameId> Builder;
Builder.build(std::move(MemProfCallStackData), MemProfFrameIndexes,
FrameHistogram);
ASSERT_THAT(Builder.getRadixArray(), testing::IsEmpty());
@@ -677,8 +677,8 @@ TEST(MemProf, RadixTreeBuilderOne) {
MemProfCallStackData.insert({llvm::memprof::hashCallStack(CS1), CS1});
llvm::DenseMap<llvm::memprof::FrameId, llvm::memprof::FrameStat>
FrameHistogram =
- llvm::memprof::computeFrameHistogram(MemProfCallStackData);
- llvm::memprof::CallStackRadixTreeBuilder Builder;
+ llvm::memprof::computeFrameHistogram<FrameId>(MemProfCallStackData);
+ llvm::memprof::CallStackRadixTreeBuilder<FrameId> Builder;
Builder.build(std::move(MemProfCallStackData), MemProfFrameIndexes,
FrameHistogram);
EXPECT_THAT(Builder.getRadixArray(), testing::ElementsAreArray({
@@ -704,8 +704,8 @@ TEST(MemProf, RadixTreeBuilderTwo) {
MemProfCallStackData.insert({llvm::memprof::hashCallStack(CS2), CS2});
llvm::DenseMap<llvm::memprof::FrameId, llvm::memprof::FrameStat>
FrameHistogram =
- llvm::memprof::computeFrameHistogram(MemProfCallStackData);
- llvm::memprof::CallStackRadixTreeBuilder Builder;
+ llvm::memprof::computeFrameHistogram<FrameId>(MemProfCallStackData);
+ llvm::memprof::CallStackRadixTreeBuilder<FrameId> Builder;
Builder.build(std::move(MemProfCallStackData), MemProfFrameIndexes,
FrameHistogram);
EXPECT_THAT(Builder.getRadixArray(),
@@ -742,8 +742,8 @@ TEST(MemProf, RadixTreeBuilderSuccessiveJumps) {
MemProfCallStackData.insert({llvm::memprof::hashCallStack(CS4), CS4});
llvm::DenseMap<llvm::memprof::FrameId, llvm::memprof::FrameStat>
FrameHistogram =
- llvm::memprof::computeFrameHistogram(MemProfCallStackData);
- llvm::memprof::CallStackRadixTreeBuilder Builder;
+ llvm::memprof::computeFrameHistogram<FrameId>(MemProfCallStackData);
+ llvm::memprof::CallStackRadixTreeBuilder<FrameId> Builder;
Builder.build(std::move(MemProfCallStackData), MemProfFrameIndexes,
FrameHistogram);
EXPECT_THAT(Builder.getRadixArray(),