6024e5c395
clang 20.0.0 (based on r547379) from build 12806354. Bug: http://b/379133546 Test: N/A Change-Id: I2eb8938af55d809de674be63cb30cf27e801862b Upstream-Commit: ad834e67b1105d15ef907f6255d4c96e8e733f57
95 lines
3.3 KiB
C++
95 lines
3.3 KiB
C++
//===--- PGOCtxProfReader.h - Contextual profile reader ---------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
///
|
|
/// Reader for contextual iFDO profile, which comes in bitstream format.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_PROFILEDATA_CTXINSTRPROFILEREADER_H
|
|
#define LLVM_PROFILEDATA_CTXINSTRPROFILEREADER_H
|
|
|
|
#include "llvm/ADT/DenseSet.h"
|
|
#include "llvm/Bitstream/BitstreamReader.h"
|
|
#include "llvm/IR/GlobalValue.h"
|
|
#include "llvm/ProfileData/PGOCtxProfWriter.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
namespace llvm {
|
|
/// A node (context) in the loaded contextual profile, suitable for mutation
|
|
/// during IPO passes. We generally expect a fraction of counters and
|
|
/// callsites to be populated. We continue to model counters as vectors, but
|
|
/// callsites are modeled as a map of a map. The expectation is that, typically,
|
|
/// there is a small number of indirect targets (usually, 1 for direct calls);
|
|
/// but potentially a large number of callsites, and, as inlining progresses,
|
|
/// the callsite count of a caller will grow.
|
|
class PGOCtxProfContext final {
|
|
public:
|
|
using CallTargetMapTy = std::map<GlobalValue::GUID, PGOCtxProfContext>;
|
|
using CallsiteMapTy = DenseMap<uint32_t, CallTargetMapTy>;
|
|
|
|
private:
|
|
friend class PGOCtxProfileReader;
|
|
GlobalValue::GUID GUID = 0;
|
|
SmallVector<uint64_t, 16> Counters;
|
|
CallsiteMapTy Callsites;
|
|
|
|
PGOCtxProfContext(GlobalValue::GUID G, SmallVectorImpl<uint64_t> &&Counters)
|
|
: GUID(G), Counters(std::move(Counters)) {}
|
|
|
|
Expected<PGOCtxProfContext &>
|
|
getOrEmplace(uint32_t Index, GlobalValue::GUID G,
|
|
SmallVectorImpl<uint64_t> &&Counters);
|
|
|
|
public:
|
|
PGOCtxProfContext(const PGOCtxProfContext &) = delete;
|
|
PGOCtxProfContext &operator=(const PGOCtxProfContext &) = delete;
|
|
PGOCtxProfContext(PGOCtxProfContext &&) = default;
|
|
PGOCtxProfContext &operator=(PGOCtxProfContext &&) = default;
|
|
|
|
GlobalValue::GUID guid() const { return GUID; }
|
|
const SmallVectorImpl<uint64_t> &counters() const { return Counters; }
|
|
const CallsiteMapTy &callsites() const { return Callsites; }
|
|
CallsiteMapTy &callsites() { return Callsites; }
|
|
|
|
bool hasCallsite(uint32_t I) const {
|
|
return Callsites.find(I) != Callsites.end();
|
|
}
|
|
|
|
const CallTargetMapTy &callsite(uint32_t I) const {
|
|
assert(hasCallsite(I) && "Callsite not found");
|
|
return Callsites.find(I)->second;
|
|
}
|
|
void getContainedGuids(DenseSet<GlobalValue::GUID> &Guids) const;
|
|
};
|
|
|
|
class PGOCtxProfileReader final {
|
|
StringRef Magic;
|
|
BitstreamCursor Cursor;
|
|
Expected<BitstreamEntry> advance();
|
|
Error readMetadata();
|
|
Error wrongValue(const Twine &);
|
|
Error unsupported(const Twine &);
|
|
|
|
Expected<std::pair<std::optional<uint32_t>, PGOCtxProfContext>>
|
|
readContext(bool ExpectIndex);
|
|
bool canReadContext();
|
|
|
|
public:
|
|
PGOCtxProfileReader(StringRef Buffer)
|
|
: Magic(Buffer.substr(0, PGOCtxProfileWriter::ContainerMagic.size())),
|
|
Cursor(Buffer.substr(PGOCtxProfileWriter::ContainerMagic.size())) {}
|
|
|
|
Expected<std::map<GlobalValue::GUID, PGOCtxProfContext>> loadContexts();
|
|
};
|
|
} // namespace llvm
|
|
#endif
|