mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-10-19 19:12:56 +02:00
7ba22bc99a
Summary: This patch extends the 'show' and 'merge' commands in llvm-profdata to handle sample PGO formats. Using the 'merge' command it is now possible to convert one sample PGO format to another. The only format that is currently not working is 'gcc'. I still need to implement support for it in lib/ProfileData. The changes in the sample profile support classes are needed for the merge operation. Reviewers: bogner Subscribers: llvm-commits Differential Revision: http://reviews.llvm.org/D6065 llvm-svn: 221032
111 lines
3.5 KiB
C++
111 lines
3.5 KiB
C++
//===- SampleProfWriter.h - Write LLVM sample profile data ----------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains definitions needed for writing sample profiles.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#ifndef LLVM_PROFILEDATA_SAMPLEPROFWRITER_H
|
|
#define LLVM_PROFILEDATA_SAMPLEPROFWRITER_H
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/ProfileData/SampleProf.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
namespace llvm {
|
|
|
|
namespace sampleprof {
|
|
|
|
enum SampleProfileFormat { SPF_None = 0, SPF_Text, SPF_Binary, SPF_GCC };
|
|
|
|
/// \brief Sample-based profile writer. Base class.
|
|
class SampleProfileWriter {
|
|
public:
|
|
SampleProfileWriter(StringRef Filename, std::error_code &EC,
|
|
sys::fs::OpenFlags Flags)
|
|
: OS(Filename, EC, Flags) {}
|
|
virtual ~SampleProfileWriter() {}
|
|
|
|
/// \brief Write sample profiles in \p S for function \p FName.
|
|
///
|
|
/// \returns true if the file was updated successfully. False, otherwise.
|
|
virtual bool write(StringRef FName, const FunctionSamples &S) = 0;
|
|
|
|
/// \brief Write sample profiles in \p S for function \p F.
|
|
bool write(const Function &F, const FunctionSamples &S) {
|
|
return write(F.getName(), S);
|
|
}
|
|
|
|
/// \brief Write all the sample profiles for all the functions in \p M.
|
|
///
|
|
/// \returns true if the file was updated successfully. False, otherwise.
|
|
bool write(const Module &M, StringMap<FunctionSamples> &P) {
|
|
for (const auto &F : M) {
|
|
StringRef Name = F.getName();
|
|
if (!write(Name, P[Name]))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/// \brief Write all the sample profiles in the given map of samples.
|
|
///
|
|
/// \returns true if the file was updated successfully. False, otherwise.
|
|
bool write(StringMap<FunctionSamples> &ProfileMap) {
|
|
for (auto &I : ProfileMap) {
|
|
StringRef FName = I.first();
|
|
FunctionSamples &Profile = I.second;
|
|
if (!write(FName, Profile))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/// \brief Profile writer factory. Create a new writer based on the value of
|
|
/// \p Format.
|
|
static std::error_code create(StringRef Filename,
|
|
std::unique_ptr<SampleProfileWriter> &Result,
|
|
SampleProfileFormat Format);
|
|
|
|
protected:
|
|
/// \brief Output stream where to emit the profile to.
|
|
raw_fd_ostream OS;
|
|
};
|
|
|
|
/// \brief Sample-based profile writer (text format).
|
|
class SampleProfileWriterText : public SampleProfileWriter {
|
|
public:
|
|
SampleProfileWriterText(StringRef F, std::error_code &EC)
|
|
: SampleProfileWriter(F, EC, sys::fs::F_Text) {}
|
|
|
|
bool write(StringRef FName, const FunctionSamples &S) override;
|
|
bool write(const Module &M, StringMap<FunctionSamples> &P) {
|
|
return SampleProfileWriter::write(M, P);
|
|
}
|
|
};
|
|
|
|
/// \brief Sample-based profile writer (binary format).
|
|
class SampleProfileWriterBinary : public SampleProfileWriter {
|
|
public:
|
|
SampleProfileWriterBinary(StringRef F, std::error_code &EC);
|
|
|
|
bool write(StringRef F, const FunctionSamples &S) override;
|
|
bool write(const Module &M, StringMap<FunctionSamples> &P) {
|
|
return SampleProfileWriter::write(M, P);
|
|
}
|
|
};
|
|
|
|
} // End namespace sampleprof
|
|
|
|
} // End namespace llvm
|
|
|
|
#endif // LLVM_PROFILEDATA_SAMPLEPROFWRITER_H
|