You've already forked linux-packaging-mono
acceptance-tests
data
debian
docs
external
Newtonsoft.Json
api-doc-tools
api-snapshot
aspnetwebstack
binary-reference-assemblies
bockbuild
boringssl
cecil
cecil-legacy
corefx
corert
helix-binaries
ikdasm
ikvm
illinker-test-assets
linker
llvm
bindings
cmake
docs
examples
include
lib
projects
resources
runtimes
scripts
test
tools
unittests
ADT
Analysis
AsmParser
BinaryFormat
Bitcode
CodeGen
DebugInfo
ExecutionEngine
FuzzMutate
IR
LineEditor
Linker
MC
MI
Object
ObjectYAML
Option
ProfileData
CMakeLists.txt
CoverageMappingTest.cpp
InstrProfTest.cpp
SampleProfTest.cpp
Support
Target
Transforms
XRay
tools
CMakeLists.txt
utils
.arcconfig
.clang-format
.clang-tidy
.gitattributes
.gitignore
CMakeLists.txt
CODE_OWNERS.TXT
CREDITS.TXT
LICENSE.TXT
LLVMBuild.txt
README.txt
RELEASE_TESTERS.TXT
configure
llvm.spec.in
nuget-buildtasks
nunit-lite
roslyn-binaries
rx
xunit-binaries
how-to-bump-roslyn-binaries.md
ikvm-native
libgc
llvm
m4
man
mcs
mk
mono
msvc
po
runtime
samples
scripts
support
tools
COPYING.LIB
LICENSE
Makefile.am
Makefile.in
NEWS
README.md
acinclude.m4
aclocal.m4
autogen.sh
code_of_conduct.md
compile
config.guess
config.h.in
config.rpath
config.sub
configure.REMOVED.git-id
configure.ac.REMOVED.git-id
depcomp
install-sh
ltmain.sh.REMOVED.git-id
missing
mkinstalldirs
mono-uninstalled.pc.in
test-driver
winconfig.h
196 lines
6.3 KiB
C++
196 lines
6.3 KiB
C++
![]() |
//===- unittest/ProfileData/SampleProfTest.cpp ------------------*- C++ -*-===//
|
||
|
//
|
||
|
// The LLVM Compiler Infrastructure
|
||
|
//
|
||
|
// This file is distributed under the University of Illinois Open Source
|
||
|
// License. See LICENSE.TXT for details.
|
||
|
//
|
||
|
//===----------------------------------------------------------------------===//
|
||
|
|
||
|
#include "llvm/ProfileData/SampleProf.h"
|
||
|
#include "llvm/ADT/StringMap.h"
|
||
|
#include "llvm/ADT/StringRef.h"
|
||
|
#include "llvm/IR/LLVMContext.h"
|
||
|
#include "llvm/IR/Metadata.h"
|
||
|
#include "llvm/IR/Module.h"
|
||
|
#include "llvm/ProfileData/SampleProfReader.h"
|
||
|
#include "llvm/ProfileData/SampleProfWriter.h"
|
||
|
#include "llvm/Support/Casting.h"
|
||
|
#include "llvm/Support/ErrorOr.h"
|
||
|
#include "llvm/Support/MemoryBuffer.h"
|
||
|
#include "llvm/Support/raw_ostream.h"
|
||
|
#include "gtest/gtest.h"
|
||
|
#include <string>
|
||
|
#include <vector>
|
||
|
|
||
|
using namespace llvm;
|
||
|
using namespace sampleprof;
|
||
|
|
||
|
static ::testing::AssertionResult NoError(std::error_code EC) {
|
||
|
if (!EC)
|
||
|
return ::testing::AssertionSuccess();
|
||
|
return ::testing::AssertionFailure() << "error " << EC.value() << ": "
|
||
|
<< EC.message();
|
||
|
}
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
struct SampleProfTest : ::testing::Test {
|
||
|
std::string Data;
|
||
|
LLVMContext Context;
|
||
|
std::unique_ptr<raw_ostream> OS;
|
||
|
std::unique_ptr<SampleProfileWriter> Writer;
|
||
|
std::unique_ptr<SampleProfileReader> Reader;
|
||
|
|
||
|
SampleProfTest()
|
||
|
: Data(), OS(new raw_string_ostream(Data)), Writer(), Reader() {}
|
||
|
|
||
|
void createWriter(SampleProfileFormat Format) {
|
||
|
auto WriterOrErr = SampleProfileWriter::create(OS, Format);
|
||
|
ASSERT_TRUE(NoError(WriterOrErr.getError()));
|
||
|
Writer = std::move(WriterOrErr.get());
|
||
|
}
|
||
|
|
||
|
void readProfile(std::unique_ptr<MemoryBuffer> &Profile) {
|
||
|
auto ReaderOrErr = SampleProfileReader::create(Profile, Context);
|
||
|
ASSERT_TRUE(NoError(ReaderOrErr.getError()));
|
||
|
Reader = std::move(ReaderOrErr.get());
|
||
|
}
|
||
|
|
||
|
void testRoundTrip(SampleProfileFormat Format) {
|
||
|
createWriter(Format);
|
||
|
|
||
|
StringRef FooName("_Z3fooi");
|
||
|
FunctionSamples FooSamples;
|
||
|
FooSamples.setName(FooName);
|
||
|
FooSamples.addTotalSamples(7711);
|
||
|
FooSamples.addHeadSamples(610);
|
||
|
FooSamples.addBodySamples(1, 0, 610);
|
||
|
FooSamples.addBodySamples(2, 0, 600);
|
||
|
FooSamples.addBodySamples(4, 0, 60000);
|
||
|
FooSamples.addBodySamples(8, 0, 60351);
|
||
|
FooSamples.addBodySamples(10, 0, 605);
|
||
|
|
||
|
StringRef BarName("_Z3bari");
|
||
|
FunctionSamples BarSamples;
|
||
|
BarSamples.setName(BarName);
|
||
|
BarSamples.addTotalSamples(20301);
|
||
|
BarSamples.addHeadSamples(1437);
|
||
|
BarSamples.addBodySamples(1, 0, 1437);
|
||
|
|
||
|
StringMap<FunctionSamples> Profiles;
|
||
|
Profiles[FooName] = std::move(FooSamples);
|
||
|
Profiles[BarName] = std::move(BarSamples);
|
||
|
|
||
|
std::error_code EC;
|
||
|
EC = Writer->write(Profiles);
|
||
|
ASSERT_TRUE(NoError(EC));
|
||
|
|
||
|
Writer->getOutputStream().flush();
|
||
|
|
||
|
auto Profile = MemoryBuffer::getMemBufferCopy(Data);
|
||
|
readProfile(Profile);
|
||
|
|
||
|
EC = Reader->read();
|
||
|
ASSERT_TRUE(NoError(EC));
|
||
|
|
||
|
StringMap<FunctionSamples> &ReadProfiles = Reader->getProfiles();
|
||
|
ASSERT_EQ(2u, ReadProfiles.size());
|
||
|
|
||
|
FunctionSamples &ReadFooSamples = ReadProfiles[FooName];
|
||
|
ASSERT_EQ(7711u, ReadFooSamples.getTotalSamples());
|
||
|
ASSERT_EQ(610u, ReadFooSamples.getHeadSamples());
|
||
|
|
||
|
FunctionSamples &ReadBarSamples = ReadProfiles[BarName];
|
||
|
ASSERT_EQ(20301u, ReadBarSamples.getTotalSamples());
|
||
|
ASSERT_EQ(1437u, ReadBarSamples.getHeadSamples());
|
||
|
|
||
|
auto VerifySummary = [](ProfileSummary &Summary) mutable {
|
||
|
ASSERT_EQ(ProfileSummary::PSK_Sample, Summary.getKind());
|
||
|
ASSERT_EQ(123603u, Summary.getTotalCount());
|
||
|
ASSERT_EQ(6u, Summary.getNumCounts());
|
||
|
ASSERT_EQ(2u, Summary.getNumFunctions());
|
||
|
ASSERT_EQ(1437u, Summary.getMaxFunctionCount());
|
||
|
ASSERT_EQ(60351u, Summary.getMaxCount());
|
||
|
|
||
|
uint32_t Cutoff = 800000;
|
||
|
auto Predicate = [&Cutoff](const ProfileSummaryEntry &PE) {
|
||
|
return PE.Cutoff == Cutoff;
|
||
|
};
|
||
|
std::vector<ProfileSummaryEntry> &Details = Summary.getDetailedSummary();
|
||
|
auto EightyPerc = find_if(Details, Predicate);
|
||
|
Cutoff = 900000;
|
||
|
auto NinetyPerc = find_if(Details, Predicate);
|
||
|
Cutoff = 950000;
|
||
|
auto NinetyFivePerc = find_if(Details, Predicate);
|
||
|
Cutoff = 990000;
|
||
|
auto NinetyNinePerc = find_if(Details, Predicate);
|
||
|
ASSERT_EQ(60000u, EightyPerc->MinCount);
|
||
|
ASSERT_EQ(60000u, NinetyPerc->MinCount);
|
||
|
ASSERT_EQ(60000u, NinetyFivePerc->MinCount);
|
||
|
ASSERT_EQ(610u, NinetyNinePerc->MinCount);
|
||
|
};
|
||
|
|
||
|
ProfileSummary &Summary = Reader->getSummary();
|
||
|
VerifySummary(Summary);
|
||
|
|
||
|
// Test that conversion of summary to and from Metadata works.
|
||
|
Metadata *MD = Summary.getMD(Context);
|
||
|
ASSERT_TRUE(MD);
|
||
|
ProfileSummary *PS = ProfileSummary::getFromMD(MD);
|
||
|
ASSERT_TRUE(PS);
|
||
|
VerifySummary(*PS);
|
||
|
delete PS;
|
||
|
|
||
|
// Test that summary can be attached to and read back from module.
|
||
|
Module M("my_module", Context);
|
||
|
M.setProfileSummary(MD);
|
||
|
MD = M.getProfileSummary();
|
||
|
ASSERT_TRUE(MD);
|
||
|
PS = ProfileSummary::getFromMD(MD);
|
||
|
ASSERT_TRUE(PS);
|
||
|
VerifySummary(*PS);
|
||
|
delete PS;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
TEST_F(SampleProfTest, roundtrip_text_profile) {
|
||
|
testRoundTrip(SampleProfileFormat::SPF_Text);
|
||
|
}
|
||
|
|
||
|
TEST_F(SampleProfTest, roundtrip_binary_profile) {
|
||
|
testRoundTrip(SampleProfileFormat::SPF_Binary);
|
||
|
}
|
||
|
|
||
|
TEST_F(SampleProfTest, sample_overflow_saturation) {
|
||
|
const uint64_t Max = std::numeric_limits<uint64_t>::max();
|
||
|
sampleprof_error Result;
|
||
|
|
||
|
StringRef FooName("_Z3fooi");
|
||
|
FunctionSamples FooSamples;
|
||
|
Result = FooSamples.addTotalSamples(1);
|
||
|
ASSERT_EQ(Result, sampleprof_error::success);
|
||
|
|
||
|
Result = FooSamples.addHeadSamples(1);
|
||
|
ASSERT_EQ(Result, sampleprof_error::success);
|
||
|
|
||
|
Result = FooSamples.addBodySamples(10, 0, 1);
|
||
|
ASSERT_EQ(Result, sampleprof_error::success);
|
||
|
|
||
|
Result = FooSamples.addTotalSamples(Max);
|
||
|
ASSERT_EQ(Result, sampleprof_error::counter_overflow);
|
||
|
ASSERT_EQ(FooSamples.getTotalSamples(), Max);
|
||
|
|
||
|
Result = FooSamples.addHeadSamples(Max);
|
||
|
ASSERT_EQ(Result, sampleprof_error::counter_overflow);
|
||
|
ASSERT_EQ(FooSamples.getHeadSamples(), Max);
|
||
|
|
||
|
Result = FooSamples.addBodySamples(10, 0, Max);
|
||
|
ASSERT_EQ(Result, sampleprof_error::counter_overflow);
|
||
|
ErrorOr<uint64_t> BodySamples = FooSamples.findSamplesAt(10, 0);
|
||
|
ASSERT_FALSE(BodySamples.getError());
|
||
|
ASSERT_EQ(BodySamples.get(), Max);
|
||
|
}
|
||
|
|
||
|
} // end anonymous namespace
|