6024e5c395
clang 20.0.0 (based on r547379) from build 12806354. Bug: http://b/379133546 Test: N/A Change-Id: I2eb8938af55d809de674be63cb30cf27e801862b Upstream-Commit: ad834e67b1105d15ef907f6255d4c96e8e733f57
67 lines
2.4 KiB
C++
67 lines
2.4 KiB
C++
//===--- VTuneSupportPlugin.h -- Support for VTune profiler ---*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Handles support for registering code with VIntel Tune's Amplifier JIT API.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_EXECUTIONENGINE_ORC_DEBUGGING_VTUNESUPPORT_H
|
|
#define LLVM_EXECUTIONENGINE_ORC_DEBUGGING_VTUNESUPPORT_H
|
|
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ExecutionEngine/Orc/Core.h"
|
|
#include "llvm/ExecutionEngine/Orc/ObjectLinkingLayer.h"
|
|
#include "llvm/ExecutionEngine/Orc/Shared/SimplePackedSerialization.h"
|
|
#include "llvm/ExecutionEngine/Orc/Shared/VTuneSharedStructs.h"
|
|
|
|
namespace llvm {
|
|
|
|
namespace orc {
|
|
|
|
class VTuneSupportPlugin : public ObjectLinkingLayer::Plugin {
|
|
public:
|
|
VTuneSupportPlugin(ExecutorProcessControl &EPC, ExecutorAddr RegisterImplAddr,
|
|
ExecutorAddr UnregisterImplAddr, bool EmitDebugInfo)
|
|
: EPC(EPC), RegisterVTuneImplAddr(RegisterImplAddr),
|
|
UnregisterVTuneImplAddr(UnregisterImplAddr),
|
|
EmitDebugInfo(EmitDebugInfo) {}
|
|
|
|
void modifyPassConfig(MaterializationResponsibility &MR,
|
|
jitlink::LinkGraph &G,
|
|
jitlink::PassConfiguration &Config) override;
|
|
|
|
Error notifyEmitted(MaterializationResponsibility &MR) override;
|
|
Error notifyFailed(MaterializationResponsibility &MR) override;
|
|
Error notifyRemovingResources(JITDylib &JD, ResourceKey K) override;
|
|
void notifyTransferringResources(JITDylib &JD, ResourceKey DstKey,
|
|
ResourceKey SrcKey) override;
|
|
|
|
static Expected<std::unique_ptr<VTuneSupportPlugin>>
|
|
Create(ExecutorProcessControl &EPC, JITDylib &JD, bool EmitDebugInfo,
|
|
bool TestMode = false);
|
|
|
|
private:
|
|
ExecutorProcessControl &EPC;
|
|
ExecutorAddr RegisterVTuneImplAddr;
|
|
ExecutorAddr UnregisterVTuneImplAddr;
|
|
std::mutex PluginMutex;
|
|
uint64_t NextMethodID = 0;
|
|
DenseMap<MaterializationResponsibility *, std::pair<uint64_t, uint64_t>>
|
|
PendingMethodIDs;
|
|
DenseMap<ResourceKey, SmallVector<std::pair<uint64_t, uint64_t>>>
|
|
LoadedMethodIDs;
|
|
bool EmitDebugInfo;
|
|
};
|
|
|
|
} // end namespace orc
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif
|