19#include "llvm/Config/config.h"
37#define DEBUG_TYPE "amplifier-jit-event-listener"
41class IntelIttnotifyInfo {
43 std::vector<std::string> SectionNamesVector;
44 std::vector<__itt_section_info> SectionInfoVector;
45 __itt_module_object *ModuleObject;
50 : ModuleObject(NULL), WrapperRef(
Wrapper){};
51 ~IntelIttnotifyInfo() {
delete ModuleObject; };
55 const char *getModuleName() {
return ModuleName.c_str(); }
57 void setModuleObject(__itt_module_object *ModuleObj) {
58 ModuleObject = ModuleObj;
61 __itt_module_object *getModuleObject() {
return ModuleObject; }
63 __itt_section_info *getSectionInfoVectorBegin() {
64 if (SectionInfoVector.size())
65 return &SectionInfoVector[0];
70 unsigned int SectionSize) {
74 int fillSectionInformation(
const ObjectFile &Obj,
77 int SectionCounter = 0;
79 for (
auto &Section : Obj.
sections()) {
80 uint64_t SectionLoadAddr = L.getSectionLoadAddress(Section);
81 if (SectionLoadAddr) {
84 __itt_section_info SectionInfo;
85 memset(&SectionInfo, 0,
sizeof(SectionInfo));
86 SectionInfo.start_addr =
reinterpret_cast<void *
>(SectionLoadAddr);
87 SectionInfo.file_offset = ElfSection.
getOffset();
88 SectionInfo.flags = ElfSection.
getFlags();
91 auto SectionNameOrError = ElfSection.
getName();
92 if (SectionNameOrError)
96 SectionInfo.size = ElfSection.
getSize();
100 if (ElfSection.
isBSS()) {
101 SectionInfo.type = itt_section_type_bss;
102 }
else if (ElfSection.
isData()) {
103 SectionInfo.type = itt_section_type_data;
104 }
else if (ElfSection.
isText()) {
105 SectionInfo.type = itt_section_type_text;
107 SectionInfoVector.push_back(SectionInfo);
114 for (
int I = 0;
I < SectionCounter; ++
I) {
115 SectionInfoVector[
I].name = SectionNamesVector[
I].c_str();
117 return SectionCounter;
124 std::unique_ptr<IntelJITEventsWrapper>
Wrapper;
125 MethodIDMap MethodIDs;
130 ObjectMap LoadedObjectMap;
131 std::map<ObjectKey, OwningBinary<ObjectFile>> DebugObjects;
133 std::map<ObjectKey, std::unique_ptr<IntelIttnotifyInfo>> KeyToIttnotify;
140 ~IntelJITEventListener() {
149static LineNumberInfo DILineInfoToIntelJITFormat(uintptr_t StartAddress,
154 Result.Offset = Address - StartAddress;
155 Result.LineNumber = Line.Line;
168 Result.method_id =
Wrapper.iJIT_GetNewMethodID();
169 Result.method_name =
const_cast<char*
>(FnName);
170 Result.method_load_address =
reinterpret_cast<void*
>(FnStart);
171 Result.method_size = FnSize;
174 Result.class_file_name = NULL;
175 Result.user_data = NULL;
176 Result.user_data_size = 0;
182int getBackwardCompatibilityMode() {
184 char *BackwardCompatibilityEnv = getenv(
"INTEL_JIT_BACKWARD_COMPATIBILITY");
185 int BackwardCompatibilityMode = 0;
186 if (BackwardCompatibilityEnv) {
190 return BackwardCompatibilityMode;
193void IntelJITEventListener::notifyObjectLoaded(
197 int BackwardCompatibilityMode = getBackwardCompatibilityMode();
198 if (BackwardCompatibilityMode == 0) {
200 std::unique_ptr<IntelIttnotifyInfo> ModuleIttnotify =
201 std::make_unique<IntelIttnotifyInfo>(*
Wrapper);
202 ModuleIttnotify->setModuleName(
208 __itt_module_object *ModuleObject =
new __itt_module_object();
209 ModuleObject->module_name = ModuleIttnotify->getModuleName();
212 ModuleObject->module_name,
213 ModuleObject->module_size);
214 ModuleObject->module_type = __itt_module_type_elf;
215 ModuleObject->section_number =
216 ModuleIttnotify->fillSectionInformation(Obj, L);
217 ModuleObject->module_buffer =
219 ModuleObject->module_id =
220 __itt_id_make((
void *)&(*ModuleObject), ModuleObject->module_size);
221 ModuleObject->section_array =
222 ModuleIttnotify->getSectionInfoVectorBegin();
223 ModuleIttnotify->setModuleObject(ModuleObject);
225 __itt_module_load_with_sections(ModuleObject);
227 KeyToIttnotify[Key] = std::move(ModuleIttnotify);
229 }
else if (BackwardCompatibilityMode == 1) {
239 MethodAddressVector Functions;
242 for (
const std::pair<SymbolRef, uint64_t> &
P :
245 std::vector<LineNumberInfo> LineInfo;
246 std::string SourceFileName;
274 auto SecOrErr =
Sym.getSection();
286 Functions.push_back((
void *)
Addr);
297 DILineInfoToIntelJITFormat((uintptr_t)
Addr, It->first, It->second));
299 if (LineInfo.size() == 0) {
312 LineInfo.push_back(last);
313 for (
size_t i = LineInfo.size() - 2; i > 0; --i)
314 LineInfo[i].LineNumber = LineInfo[i - 1].LineNumber;
316 SourceFileName = Lines.front().second.FileName;
318 const_cast<char *
>(SourceFileName.c_str());
331 LoadedObjectMap[ObjData] = Functions;
332 DebugObjects[Key] = std::move(DebugObjOwner);
336void IntelJITEventListener::notifyFreeingObject(ObjectKey Key) {
338 int BackwardCompatibilityMode = getBackwardCompatibilityMode();
339 if (BackwardCompatibilityMode == 0) {
340 if (KeyToIttnotify.find(Key) == KeyToIttnotify.end())
342 __itt_module_unload_with_sections(KeyToIttnotify[Key]->getModuleObject());
345 KeyToIttnotify[Key]->getModuleObject()->module_name,
346 KeyToIttnotify[Key]->getModuleObject()->module_size);
347 KeyToIttnotify.erase(Key);
348 }
else if (BackwardCompatibilityMode == 1) {
351 if (DebugObjects.find(Key) == DebugObjects.end())
355 const ObjectFile &DebugObj = *DebugObjects[Key].getBinary();
359 ObjectMap::iterator OI = LoadedObjectMap.find(ObjData);
360 if (OI == LoadedObjectMap.end())
362 MethodAddressVector &Functions = OI->second;
365 for (MethodAddressVector::iterator FI = Functions.begin(),
366 FE = Functions.end();
368 void *FnStart =
const_cast<void *
>(*FI);
369 MethodIDMap::iterator
MI = MethodIDs.find(FnStart);
370 if (
MI != MethodIDs.end()) {
378 LoadedObjectMap.erase(OI);
379 DebugObjects.erase(Key);
393 return new IntelJITEventListener(TestImpl);
amdgpu aa AMDGPU Address space based Alias Analysis Wrapper
This file defines the DenseMap class.
static std::unique_ptr< DWARFContext > create(const object::ObjectFile &Obj, ProcessDebugRelocations RelocAction=ProcessDebugRelocations::Process, const LoadedObjectInfo *L=nullptr, std::string DWPName="", std::function< void(Error)> RecoverableErrorHandler=WithColor::defaultErrorHandler, std::function< void(Error)> WarningHandler=WithColor::defaultWarningHandler, bool ThreadSafe=false)
Tagged union holding either a T or a Error.
Error takeError()
Take ownership of the stored error.
int iJitIttNotifyInfo(IttEventType EventType, const char *Name, unsigned int Size)
JITEventListener - Abstract interface for use by the JIT to notify clients about significant events d...
virtual void notifyFreeingObject(ObjectKey K)
notifyFreeingObject - Called just before the memory associated with a previously emitted object is re...
virtual void notifyObjectLoaded(ObjectKey K, const object::ObjectFile &Obj, const RuntimeDyld::LoadedObjectInfo &L)
notifyObjectLoaded - Called after an object has had its sections allocated and addresses assigned to ...
static JITEventListener * createIntelJITEventListener()
size_t getBufferSize() const
const char * getBufferStart() const
StringRef getBuffer() const
Information about the loaded object.
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
StringRef - Represent a constant reference to a string, i.e.
bool getAsInteger(unsigned Radix, T &Result) const
Parse the current string as an integer of the specified radix.
std::string str() const
str - Get the contents as an std::string.
constexpr const char * data() const
data - Get a pointer to the start of the string (which may not be null terminated).
StringRef getData() const
MemoryBufferRef getMemoryBufferRef() const
uint64_t getFlags() const
uint64_t getOffset() const
This class is the base class for all object file types.
virtual section_iterator section_end() const =0
section_iterator_range sections() const
uint64_t getIndex() const
bool isData() const
Whether this section contains data, not instructions.
bool isBSS() const
Whether this section contains BSS uninitialized data.
bool isText() const
Whether this section contains instructions.
Expected< StringRef > getName() const
This is a value type class that represents a single symbol in the list of symbols in the object file.
LLVMJITEventListenerRef LLVMCreateIntelJITEventListener(void)
struct LLVMOpaqueJITEventListener * LLVMJITEventListenerRef
@ iJVM_EVENT_TYPE_METHOD_LOAD_FINISHED
@ iJVM_EVENT_TYPE_METHOD_UNLOAD_START
std::vector< std::pair< SymbolRef, uint64_t > > computeSymbolSizes(const ObjectFile &O)
uint64_t MD5Hash(const FunctionId &Obj)
This is an optimization pass for GlobalISel generic memory operations.
LLVMAttributeRef wrap(Attribute Attr)
void consumeError(Error Err)
Consume a Error without doing anything.
unsigned int line_number_size
pLineNumberInfo line_number_table
A format-neutral container for source line information.