11#include "llvm/ADT/SmallVectorExtras.h"
12#include "llvm/ADT/TypeSwitch.h"
13#include "llvm/IR/Metadata.h"
14#include "llvm/IR/Module.h"
15#include "llvm/Support/FileSystem.h"
16#include "llvm/Support/Path.h"
30 : debugEmissionIsEnabled(
false), llvmModule(llvmModule),
35 debugEmissionIsEnabled =
true;
48 llvm::DEBUG_METADATA_VERSION);
50 const llvm::Triple &targetTriple = llvmModule.getTargetTriple();
51 if (targetTriple.isKnownWindowsMSVCEnvironment()) {
55 llvmModule.addModuleFlag(llvm::Module::Warning,
kCodeViewKey, 1);
61 if (!debugEmissionIsEnabled)
69 llvmFunc.setSubprogram(
translate(spLoc.getMetadata()));
76llvm::DIType *DebugTranslation::translateImpl(DINullTypeAttr attr) {
86DebugTranslation::getExpressionAttrOrNull(DIExpressionAttr attr) {
92llvm::MDString *DebugTranslation::getMDStringOrNull(StringAttr stringAttr) {
93 if (!stringAttr || stringAttr.empty())
95 return llvm::MDString::get(llvmCtx, stringAttr);
100 if (elements.empty())
103 llvm::map_to_vector(elements, [&](
DINodeAttr attr) -> llvm::Metadata * {
104 if (DIAnnotationAttr annAttr = dyn_cast<DIAnnotationAttr>(attr)) {
105 llvm::Metadata *ops[2] = {
106 llvm::MDString::get(llvmCtx, annAttr.getName()),
107 llvm::MDString::get(llvmCtx, annAttr.getValue())};
108 return llvm::MDNode::get(llvmCtx, ops);
112 return llvm::MDNode::get(llvmCtx, llvmElements);
115llvm::DIBasicType *DebugTranslation::translateImpl(DIBasicTypeAttr attr) {
116 return llvm::DIBasicType::get(
117 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
118 attr.getSizeInBits(),
119 0, attr.getEncoding(), llvm::DINode::FlagZero);
122llvm::DICompileUnit *DebugTranslation::translateImpl(DICompileUnitAttr attr) {
123 llvm::DIBuilder builder(llvmModule);
124 return builder.createCompileUnit(
125 attr.getSourceLanguage(),
translate(attr.getFile()),
126 attr.getProducer() ? attr.getProducer().getValue() :
"",
127 attr.getIsOptimized(),
129 attr.getSplitDebugFilename() ? attr.getSplitDebugFilename().getValue()
131 static_cast<llvm::DICompileUnit::DebugEmissionKind
>(
132 attr.getEmissionKind()),
134 static_cast<llvm::DICompileUnit::DebugNameTableKind
>(
135 attr.getNameTableKind()));
140template <
class DINodeT,
class... Ts>
143 return DINodeT::getDistinct(std::forward<Ts>(args)...);
144 return DINodeT::get(std::forward<Ts>(args)...);
147llvm::TempDICompositeType
148DebugTranslation::translateTemporaryImpl(DICompositeTypeAttr attr) {
149 return llvm::DICompositeType::getTemporary(
150 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
nullptr,
151 attr.getLine(),
nullptr,
nullptr, attr.getSizeInBits(),
152 attr.getAlignInBits(),
154 static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
155 nullptr, 0, std::nullopt,
159llvm::TempDISubprogram
160DebugTranslation::translateTemporaryImpl(DISubprogramAttr attr) {
161 return llvm::DISubprogram::getTemporary(
162 llvmCtx,
nullptr, {}, {},
163 nullptr, attr.getLine(),
nullptr,
165 0, llvm::DINode::FlagZero,
166 static_cast<llvm::DISubprogram::DISPFlags
>(attr.getSubprogramFlags()),
170llvm::DICompositeType *
171DebugTranslation::translateImpl(DICompositeTypeAttr attr) {
174 bool isDistinct =
false;
175 switch (attr.getTag()) {
176 case llvm::dwarf::DW_TAG_class_type:
177 case llvm::dwarf::DW_TAG_enumeration_type:
178 case llvm::dwarf::DW_TAG_structure_type:
179 case llvm::dwarf::DW_TAG_union_type:
184 isDistinct, llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
186 translate(attr.getBaseType()), attr.getSizeInBits(),
187 attr.getAlignInBits(),
189 static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
190 getMDTupleOrNull(attr.getElements()),
191 0, std::nullopt,
nullptr,
194 getExpressionAttrOrNull(attr.getDataLocation()),
195 getExpressionAttrOrNull(attr.getAssociated()),
196 getExpressionAttrOrNull(attr.getAllocated()),
197 getExpressionAttrOrNull(attr.getRank()));
200llvm::DIDerivedType *DebugTranslation::translateImpl(DIDerivedTypeAttr attr) {
201 return llvm::DIDerivedType::get(
202 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
204 nullptr,
translate(attr.getBaseType()), attr.getSizeInBits(),
205 attr.getAlignInBits(), attr.getOffsetInBits(),
206 attr.getDwarfAddressSpace(), std::nullopt,
207 static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
211llvm::DIStringType *DebugTranslation::translateImpl(DIStringTypeAttr attr) {
212 return llvm::DIStringType::get(
213 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
215 getExpressionAttrOrNull(attr.getStringLengthExp()),
216 getExpressionAttrOrNull(attr.getStringLocationExp()),
217 attr.getSizeInBits(), attr.getAlignInBits(), attr.getEncoding());
220llvm::DIFile *DebugTranslation::translateImpl(DIFileAttr attr) {
221 return llvm::DIFile::get(llvmCtx, getMDStringOrNull(attr.getName()),
222 getMDStringOrNull(attr.getDirectory()));
225llvm::DILabel *DebugTranslation::translateImpl(DILabelAttr attr) {
226 return llvm::DILabel::get(llvmCtx,
translate(attr.getScope()),
227 getMDStringOrNull(attr.getName()),
228 translate(attr.getFile()), attr.getLine(),
233llvm::DILexicalBlock *DebugTranslation::translateImpl(DILexicalBlockAttr attr) {
234 return llvm::DILexicalBlock::getDistinct(llvmCtx,
translate(attr.getScope()),
236 attr.getLine(), attr.getColumn());
239llvm::DILexicalBlockFile *
240DebugTranslation::translateImpl(DILexicalBlockFileAttr attr) {
241 return llvm::DILexicalBlockFile::getDistinct(
243 attr.getDiscriminator());
246llvm::DILocalScope *DebugTranslation::translateImpl(DILocalScopeAttr attr) {
247 return cast<llvm::DILocalScope>(
translate(DINodeAttr(attr)));
250llvm::DIVariable *DebugTranslation::translateImpl(DIVariableAttr attr) {
251 return cast<llvm::DIVariable>(
translate(DINodeAttr(attr)));
254llvm::DILocalVariable *
255DebugTranslation::translateImpl(DILocalVariableAttr attr) {
256 return llvm::DILocalVariable::get(
257 llvmCtx,
translate(attr.getScope()), getMDStringOrNull(attr.getName()),
259 attr.getArg(),
static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
260 attr.getAlignInBits(),
264llvm::DIGlobalVariable *
265DebugTranslation::translateImpl(DIGlobalVariableAttr attr) {
266 return llvm::DIGlobalVariable::getDistinct(
267 llvmCtx,
translate(attr.getScope()), getMDStringOrNull(attr.getName()),
268 getMDStringOrNull(attr.getLinkageName()),
translate(attr.getFile()),
269 attr.getLine(),
translate(attr.getType()), attr.getIsLocalToUnit(),
270 attr.getIsDefined(),
nullptr,
nullptr, attr.getAlignInBits(),
nullptr);
274DebugTranslation::translateRecursive(DIRecursiveTypeAttrInterface attr) {
275 DistinctAttr recursiveId = attr.getRecId();
276 if (
auto *iter = recursiveNodeMap.find(recursiveId);
277 iter != recursiveNodeMap.end()) {
280 assert(!attr.getIsRecSelf() &&
"unbound DI recursive self reference");
282 auto setRecursivePlaceholder = [&](llvm::DINode *placeholder) {
283 recursiveNodeMap.try_emplace(recursiveId, placeholder);
288 .Case([&](DICompositeTypeAttr attr) {
289 auto temporary = translateTemporaryImpl(attr);
290 setRecursivePlaceholder(temporary.get());
293 auto *concrete = translateImpl(attr);
294 temporary->replaceAllUsesWith(concrete);
297 .Case([&](DISubprogramAttr attr) {
298 auto temporary = translateTemporaryImpl(attr);
299 setRecursivePlaceholder(temporary.get());
302 auto *concrete = translateImpl(attr);
303 temporary->replaceAllUsesWith(concrete);
307 assert(recursiveNodeMap.back().first == recursiveId &&
308 "internal inconsistency: unexpected recursive translation stack");
309 recursiveNodeMap.pop_back();
314llvm::DIScope *DebugTranslation::translateImpl(DIScopeAttr attr) {
315 return cast<llvm::DIScope>(
translate(DINodeAttr(attr)));
318llvm::DISubprogram *DebugTranslation::translateImpl(DISubprogramAttr attr) {
319 if (
auto iter = distinctAttrToNode.find(attr.getId());
320 iter != distinctAttrToNode.end())
321 return cast<llvm::DISubprogram>(iter->second);
323 llvm::DIScope *scope =
translate(attr.getScope());
324 llvm::DIFile *file =
translate(attr.getFile());
325 llvm::DIType *type =
translate(attr.getType());
326 llvm::DICompileUnit *compileUnit =
translate(attr.getCompileUnit());
330 if (
auto iter = distinctAttrToNode.find(attr.getId());
331 iter != distinctAttrToNode.end())
332 return cast<llvm::DISubprogram>(iter->second);
334 bool isDefinition =
static_cast<bool>(attr.getSubprogramFlags() &
335 LLVM::DISubprogramFlags::Definition);
338 isDefinition, llvmCtx, scope, getMDStringOrNull(attr.getName()),
339 getMDStringOrNull(attr.getLinkageName()), file, attr.getLine(), type,
342 0, llvm::DINode::FlagZero,
343 static_cast<llvm::DISubprogram::DISPFlags
>(attr.getSubprogramFlags()),
344 compileUnit,
nullptr,
nullptr,
345 getMDTupleOrNull(attr.getRetainedNodes()),
nullptr,
346 getMDTupleOrNull(attr.getAnnotations()));
348 distinctAttrToNode.try_emplace(attr.getId(), node);
352llvm::DIModule *DebugTranslation::translateImpl(DIModuleAttr attr) {
353 return llvm::DIModule::get(
355 getMDStringOrNull(attr.getName()),
356 getMDStringOrNull(attr.getConfigMacros()),
357 getMDStringOrNull(attr.getIncludePath()),
358 getMDStringOrNull(attr.getApinotes()), attr.getLine(), attr.getIsDecl());
361llvm::DINamespace *DebugTranslation::translateImpl(DINamespaceAttr attr) {
362 return llvm::DINamespace::get(llvmCtx,
translate(attr.getScope()),
363 getMDStringOrNull(attr.getName()),
364 attr.getExportSymbols());
367llvm::DIImportedEntity *
368DebugTranslation::translateImpl(DIImportedEntityAttr attr) {
369 return llvm::DIImportedEntity::get(
370 llvmCtx, attr.getTag(),
translate(attr.getScope()),
372 getMDStringOrNull(attr.getName()), getMDTupleOrNull(attr.getElements()));
375llvm::DISubrange *DebugTranslation::translateImpl(DISubrangeAttr attr) {
376 auto getMetadataOrNull = [&](Attribute attr) -> llvm::Metadata * {
380 llvm::Metadata *metadata =
381 llvm::TypeSwitch<Attribute, llvm::Metadata *>(attr)
382 .Case([&](IntegerAttr intAttr) {
383 return llvm::ConstantAsMetadata::get(llvm::ConstantInt::getSigned(
384 llvm::Type::getInt64Ty(llvmCtx), intAttr.getInt()));
386 .Case([&](LLVM::DIExpressionAttr expr) {
389 .Case([&](LLVM::DILocalVariableAttr local) {
392 .Case<>([&](LLVM::DIGlobalVariableAttr global) {
398 return llvm::DISubrange::get(llvmCtx, getMetadataOrNull(attr.getCount()),
399 getMetadataOrNull(attr.getLowerBound()),
400 getMetadataOrNull(attr.getUpperBound()),
401 getMetadataOrNull(attr.getStride()));
404llvm::DICommonBlock *DebugTranslation::translateImpl(DICommonBlockAttr attr) {
405 return llvm::DICommonBlock::get(llvmCtx,
translate(attr.getScope()),
407 getMDStringOrNull(attr.getName()),
408 translate(attr.getFile()), attr.getLine());
411llvm::DIGenericSubrange *
412DebugTranslation::translateImpl(DIGenericSubrangeAttr attr) {
413 auto getMetadataOrNull = [&](Attribute attr) -> llvm::Metadata * {
417 llvm::Metadata *metadata =
418 llvm::TypeSwitch<Attribute, llvm::Metadata *>(attr)
419 .Case([&](LLVM::DIExpressionAttr expr) {
422 .Case([&](LLVM::DILocalVariableAttr local) {
425 .Case([&](LLVM::DIGlobalVariableAttr global) {
431 return llvm::DIGenericSubrange::get(llvmCtx,
432 getMetadataOrNull(attr.getCount()),
433 getMetadataOrNull(attr.getLowerBound()),
434 getMetadataOrNull(attr.getUpperBound()),
435 getMetadataOrNull(attr.getStride()));
438llvm::DISubroutineType *
439DebugTranslation::translateImpl(DISubroutineTypeAttr attr) {
441 SmallVector<llvm::Metadata *> types;
442 for (DITypeAttr type : attr.getTypes())
444 return llvm::DISubroutineType::get(
445 llvmCtx, llvm::DINode::FlagZero, attr.getCallingConvention(),
446 llvm::DITypeArray(llvm::MDNode::get(llvmCtx, types)));
449llvm::DIType *DebugTranslation::translateImpl(DITypeAttr attr) {
450 return cast<llvm::DIType>(
translate(DINodeAttr(attr)));
457 if (llvm::DINode *node = attrToNode.lookup(attr))
460 llvm::DINode *node =
nullptr;
463 if (
auto recTypeAttr = dyn_cast<DIRecursiveTypeAttrInterface>(attr))
464 if (recTypeAttr.getRecId())
465 node = translateRecursive(recTypeAttr);
469 .Case<DIBasicTypeAttr, DICommonBlockAttr, DICompileUnitAttr,
470 DICompositeTypeAttr, DIDerivedTypeAttr, DIFileAttr,
471 DIGenericSubrangeAttr, DIGlobalVariableAttr,
472 DIImportedEntityAttr, DILabelAttr, DILexicalBlockAttr,
473 DILexicalBlockFileAttr, DILocalVariableAttr, DIModuleAttr,
474 DINamespaceAttr, DINullTypeAttr, DIStringTypeAttr,
475 DISubprogramAttr, DISubrangeAttr, DISubroutineTypeAttr>(
476 [&](
auto attr) {
return translateImpl(attr); });
478 if (node && !node->isTemporary())
479 attrToNode.insert({attr, node});
489 llvm::DILocalScope *scope) {
490 if (!debugEmissionIsEnabled)
500 for (
const DIExpressionElemAttr &op : attr.getOperations()) {
501 ops.push_back(op.getOpcode());
502 append_range(ops, op.getArguments());
505 return llvm::DIExpression::get(llvmCtx, ops);
508llvm::DIGlobalVariableExpression *
510 LLVM::DIGlobalVariableExpressionAttr attr) {
511 return llvm::DIGlobalVariableExpression::get(
517 llvm::DILocalScope *scope,
518 llvm::DILocation *inlinedAt) {
520 if (isa<UnknownLoc>(loc))
524 auto existingIt = locationToLoc.find(std::make_tuple(loc, scope, inlinedAt));
525 if (existingIt != locationToLoc.end())
526 return existingIt->second;
528 llvm::DILocation *llvmLoc =
nullptr;
529 if (
auto callLoc = dyn_cast<CallSiteLoc>(loc)) {
531 auto *callerLoc =
translateLoc(callLoc.getCaller(), scope, inlinedAt);
539 callerLoc = inlinedAt;
541 llvmLoc =
translateLoc(callLoc.getCallee(),
nullptr, callerLoc);
546 }
else if (
auto fileLoc = dyn_cast<FileLineColLoc>(loc)) {
551 llvm::DILocation::get(llvmCtx, fileLoc.getLine(), fileLoc.getColumn(),
552 scope,
const_cast<llvm::DILocation *
>(inlinedAt));
554 }
else if (
auto fusedLoc = dyn_cast<FusedLoc>(loc)) {
558 if (
auto scopedAttr =
559 dyn_cast_or_null<LLVM::DILocalScopeAttr>(fusedLoc.getMetadata()))
563 llvmLoc =
translateLoc(locations.front(), scope, inlinedAt);
564 for (
Location locIt : locations.drop_front()) {
565 llvmLoc = llvm::DILocation::getMergedLocation(
569 }
else if (
auto nameLoc = dyn_cast<NameLoc>(loc)) {
570 llvmLoc =
translateLoc(nameLoc.getChildLoc(), scope, inlinedAt);
572 }
else if (
auto opaqueLoc = dyn_cast<OpaqueLoc>(loc)) {
573 llvmLoc =
translateLoc(opaqueLoc.getFallbackLocation(), scope, inlinedAt);
575 llvm_unreachable(
"unknown location kind");
578 locationToLoc.try_emplace(std::make_tuple(loc, scope, inlinedAt), llvmLoc);
583llvm::DIFile *DebugTranslation::translateFile(StringRef fileName) {
584 auto *&file = fileMap[fileName];
589 if (currentWorkingDir.empty())
590 llvm::sys::fs::current_path(currentWorkingDir);
592 StringRef directory = currentWorkingDir;
593 SmallString<128> dirBuf;
594 SmallString<128> fileBuf;
595 if (llvm::sys::path::is_absolute(fileName)) {
598 auto fileIt = llvm::sys::path::begin(fileName);
599 auto fileE = llvm::sys::path::end(fileName);
600 auto curDirIt = llvm::sys::path::begin(directory);
601 auto curDirE = llvm::sys::path::end(directory);
602 for (; curDirIt != curDirE && *curDirIt == *fileIt; ++curDirIt, ++fileIt)
603 llvm::sys::path::append(dirBuf, *curDirIt);
604 if (std::distance(llvm::sys::path::begin(directory), curDirIt) == 1) {
607 directory = StringRef();
609 for (; fileIt != fileE; ++fileIt)
610 llvm::sys::path::append(fileBuf, *fileIt);
615 return (file = llvm::DIFile::get(llvmCtx, fileName, directory));
static constexpr StringRef kDebugVersionKey
static WalkResult interruptIfValidLocation(Operation *op)
A utility walker that interrupts if the operation has valid debug information.
static constexpr StringRef kCodeViewKey
static DINodeT * getDistinctOrUnique(bool isDistinct, Ts &&...args)
Returns a new DINodeT that is either distinct or not, depending on isDistinct.
false
Parses a map_entries map type from a string format back into its numeric value.
This class represents a fused location whose metadata is known to be an instance of the given type.
This class represents the base attribute for all debug info attributes.
void translate(LLVMFuncOp func, llvm::Function &llvmFunc)
Translate the debug information for the given function.
llvm::DIExpression * translateExpression(LLVM::DIExpressionAttr attr)
Translates the given DWARF expression metadata to to LLVM.
void addModuleFlagsIfNotPresent()
Adds the necessary module flags to the module, if not yet present.
DebugTranslation(Operation *module, llvm::Module &llvmModule)
llvm::DILocation * translateLoc(Location loc, llvm::DILocalScope *scope)
Translate the given location to an llvm debug location.
llvm::DIGlobalVariableExpression * translateGlobalVariableExpression(LLVM::DIGlobalVariableExpressionAttr attr)
Translates the given DWARF global variable expression to LLVM.
This class defines the main interface for locations in MLIR and acts as a non-nullable wrapper around...
Operation is the basic unit of execution within MLIR.
Location getLoc()
The source location the operation was defined or derived from.
std::enable_if_t< llvm::function_traits< std::decay_t< FnT > >::num_args==1, RetT > walk(FnT &&callback)
Walk the operation by calling the callback for each nested operation (including this one),...
A utility result that is used to signal how to proceed with an ongoing walk:
static WalkResult advance()
static WalkResult interrupt()
Include the generated interface declarations.
llvm::TypeSwitch< T, ResultT > TypeSwitch