11 #include "llvm/ADT/TypeSwitch.h"
12 #include "llvm/IR/Metadata.h"
13 #include "llvm/IR/Module.h"
14 #include "llvm/Support/FileSystem.h"
15 #include "llvm/Support/Path.h"
29 : debugEmissionIsEnabled(false), llvmModule(llvmModule),
34 debugEmissionIsEnabled =
true;
40 StringRef debugVersionKey =
"Debug Info Version";
41 if (!llvmModule.getModuleFlag(debugVersionKey))
42 llvmModule.addModuleFlag(llvm::Module::Warning, debugVersionKey,
43 llvm::DEBUG_METADATA_VERSION);
46 LLVM::LLVMDialect::getTargetTripleAttrName())) {
48 llvm::Triple(cast<StringAttr>(targetTripleAttr).getValue());
49 if (targetTriple.isKnownWindowsMSVCEnvironment()) {
52 llvmModule.addModuleFlag(llvm::Module::Warning,
"CodeView", 1);
62 if (!debugEmissionIsEnabled)
70 llvmFunc.setSubprogram(
translate(spLoc.getMetadata()));
77 llvm::DIType *DebugTranslation::translateImpl(DINullTypeAttr attr) {
87 DebugTranslation::getExpressionAttrOrNull(DIExpressionAttr attr) {
93 llvm::MDString *DebugTranslation::getMDStringOrNull(StringAttr stringAttr) {
94 if (!stringAttr || stringAttr.empty())
101 if (elements.empty())
104 llvm::map_range(elements, [&](
DINodeAttr attr) -> llvm::Metadata * {
105 if (DIAnnotationAttr annAttr = dyn_cast<DIAnnotationAttr>(attr)) {
106 llvm::Metadata *ops[2] = {
116 llvm::DIBasicType *DebugTranslation::translateImpl(DIBasicTypeAttr attr) {
118 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
119 attr.getSizeInBits(),
120 0, attr.getEncoding(), llvm::DINode::FlagZero);
123 llvm::DICompileUnit *DebugTranslation::translateImpl(DICompileUnitAttr attr) {
124 llvm::DIBuilder builder(llvmModule);
125 return builder.createCompileUnit(
126 attr.getSourceLanguage(),
translate(attr.getFile()),
127 attr.getProducer() ? attr.getProducer().getValue() :
"",
128 attr.getIsOptimized(),
130 static_cast<llvm::DICompileUnit::DebugEmissionKind
>(
131 attr.getEmissionKind()),
133 static_cast<llvm::DICompileUnit::DebugNameTableKind
>(
134 attr.getNameTableKind()));
139 template <
class DINodeT,
class... Ts>
142 return DINodeT::getDistinct(std::forward<Ts>(args)...);
146 llvm::TempDICompositeType
147 DebugTranslation::translateTemporaryImpl(DICompositeTypeAttr attr) {
148 return llvm::DICompositeType::getTemporary(
149 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
nullptr,
150 attr.getLine(),
nullptr,
nullptr, attr.getSizeInBits(),
151 attr.getAlignInBits(),
153 static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
158 llvm::TempDISubprogram
159 DebugTranslation::translateTemporaryImpl(DISubprogramAttr attr) {
160 return llvm::DISubprogram::getTemporary(
161 llvmCtx,
nullptr, {}, {},
162 nullptr, attr.getLine(),
nullptr,
164 0, llvm::DINode::FlagZero,
165 static_cast<llvm::DISubprogram::DISPFlags
>(attr.getSubprogramFlags()),
169 llvm::DICompositeType *
170 DebugTranslation::translateImpl(DICompositeTypeAttr attr) {
173 bool isDistinct =
false;
174 switch (attr.getTag()) {
175 case llvm::dwarf::DW_TAG_class_type:
176 case llvm::dwarf::DW_TAG_enumeration_type:
177 case llvm::dwarf::DW_TAG_structure_type:
178 case llvm::dwarf::DW_TAG_union_type:
182 return getDistinctOrUnique<llvm::DICompositeType>(
183 isDistinct, llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
185 translate(attr.getBaseType()), attr.getSizeInBits(),
186 attr.getAlignInBits(),
188 static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
189 getMDTupleOrNull(attr.getElements()),
193 getExpressionAttrOrNull(attr.getDataLocation()),
194 getExpressionAttrOrNull(attr.getAssociated()),
195 getExpressionAttrOrNull(attr.getAllocated()),
196 getExpressionAttrOrNull(attr.getRank()));
199 llvm::DIDerivedType *DebugTranslation::translateImpl(DIDerivedTypeAttr attr) {
201 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
203 nullptr,
translate(attr.getBaseType()), attr.getSizeInBits(),
204 attr.getAlignInBits(), attr.getOffsetInBits(),
205 attr.getDwarfAddressSpace(), std::nullopt,
206 llvm::DINode::FlagZero,
translate(attr.getExtraData()));
209 llvm::DIStringType *DebugTranslation::translateImpl(DIStringTypeAttr attr) {
211 llvmCtx, attr.getTag(), getMDStringOrNull(attr.getName()),
213 getExpressionAttrOrNull(attr.getStringLengthExp()),
214 getExpressionAttrOrNull(attr.getStringLocationExp()),
215 attr.getSizeInBits(), attr.getAlignInBits(), attr.getEncoding());
218 llvm::DIFile *DebugTranslation::translateImpl(DIFileAttr attr) {
220 getMDStringOrNull(attr.getDirectory()));
223 llvm::DILabel *DebugTranslation::translateImpl(DILabelAttr attr) {
225 getMDStringOrNull(attr.getName()),
226 translate(attr.getFile()), attr.getLine());
229 llvm::DILexicalBlock *DebugTranslation::translateImpl(DILexicalBlockAttr attr) {
230 return llvm::DILexicalBlock::getDistinct(llvmCtx,
translate(attr.getScope()),
232 attr.getLine(), attr.getColumn());
235 llvm::DILexicalBlockFile *
236 DebugTranslation::translateImpl(DILexicalBlockFileAttr attr) {
237 return llvm::DILexicalBlockFile::getDistinct(
239 attr.getDiscriminator());
242 llvm::DILocalScope *DebugTranslation::translateImpl(
DILocalScopeAttr attr) {
246 llvm::DIVariable *DebugTranslation::translateImpl(
DIVariableAttr attr) {
250 llvm::DILocalVariable *
251 DebugTranslation::translateImpl(DILocalVariableAttr attr) {
253 llvmCtx,
translate(attr.getScope()), getMDStringOrNull(attr.getName()),
255 attr.getArg(),
static_cast<llvm::DINode::DIFlags
>(attr.getFlags()),
256 attr.getAlignInBits(),
260 llvm::DIGlobalVariable *
261 DebugTranslation::translateImpl(DIGlobalVariableAttr attr) {
262 return llvm::DIGlobalVariable::getDistinct(
263 llvmCtx,
translate(attr.getScope()), getMDStringOrNull(attr.getName()),
264 getMDStringOrNull(attr.getLinkageName()),
translate(attr.getFile()),
265 attr.getLine(),
translate(attr.getType()), attr.getIsLocalToUnit(),
266 attr.getIsDefined(),
nullptr,
nullptr, attr.getAlignInBits(),
nullptr);
270 DebugTranslation::translateRecursive(DIRecursiveTypeAttrInterface attr) {
272 if (
auto *iter = recursiveNodeMap.find(recursiveId);
273 iter != recursiveNodeMap.end()) {
276 assert(!attr.getIsRecSelf() &&
"unbound DI recursive self reference");
278 auto setRecursivePlaceholder = [&](llvm::DINode *placeholder) {
279 recursiveNodeMap.try_emplace(recursiveId, placeholder);
282 llvm::DINode *result =
284 .Case<DICompositeTypeAttr>([&](
auto attr) {
285 auto temporary = translateTemporaryImpl(attr);
286 setRecursivePlaceholder(temporary.get());
289 auto *concrete = translateImpl(attr);
290 temporary->replaceAllUsesWith(concrete);
293 .Case<DISubprogramAttr>([&](
auto attr) {
294 auto temporary = translateTemporaryImpl(attr);
295 setRecursivePlaceholder(temporary.get());
298 auto *concrete = translateImpl(attr);
299 temporary->replaceAllUsesWith(concrete);
303 assert(recursiveNodeMap.back().first == recursiveId &&
304 "internal inconsistency: unexpected recursive translation stack");
305 recursiveNodeMap.pop_back();
310 llvm::DIScope *DebugTranslation::translateImpl(
DIScopeAttr attr) {
314 llvm::DISubprogram *DebugTranslation::translateImpl(DISubprogramAttr attr) {
315 if (
auto iter = distinctAttrToNode.find(attr.getId());
316 iter != distinctAttrToNode.end())
317 return cast<llvm::DISubprogram>(iter->second);
319 llvm::DIScope *scope =
translate(attr.getScope());
320 llvm::DIFile *file =
translate(attr.getFile());
321 llvm::DIType *type =
translate(attr.getType());
322 llvm::DICompileUnit *compileUnit =
translate(attr.getCompileUnit());
326 if (
auto iter = distinctAttrToNode.find(attr.getId());
327 iter != distinctAttrToNode.end())
328 return cast<llvm::DISubprogram>(iter->second);
330 bool isDefinition =
static_cast<bool>(attr.getSubprogramFlags() &
331 LLVM::DISubprogramFlags::Definition);
333 llvm::DISubprogram *node = getDistinctOrUnique<llvm::DISubprogram>(
334 isDefinition, llvmCtx, scope, getMDStringOrNull(attr.getName()),
335 getMDStringOrNull(attr.getLinkageName()), file, attr.getLine(), type,
338 0, llvm::DINode::FlagZero,
339 static_cast<llvm::DISubprogram::DISPFlags
>(attr.getSubprogramFlags()),
340 compileUnit,
nullptr,
nullptr,
341 getMDTupleOrNull(attr.getRetainedNodes()),
nullptr,
342 getMDTupleOrNull(attr.getAnnotations()));
344 distinctAttrToNode.try_emplace(attr.getId(), node);
348 llvm::DIModule *DebugTranslation::translateImpl(DIModuleAttr attr) {
351 getMDStringOrNull(attr.getName()),
352 getMDStringOrNull(attr.getConfigMacros()),
353 getMDStringOrNull(attr.getIncludePath()),
354 getMDStringOrNull(attr.getApinotes()), attr.getLine(), attr.getIsDecl());
357 llvm::DINamespace *DebugTranslation::translateImpl(DINamespaceAttr attr) {
359 getMDStringOrNull(attr.getName()),
360 attr.getExportSymbols());
363 llvm::DIImportedEntity *
364 DebugTranslation::translateImpl(DIImportedEntityAttr attr) {
366 llvmCtx, attr.getTag(),
translate(attr.getScope()),
368 getMDStringOrNull(attr.getName()), getMDTupleOrNull(attr.getElements()));
371 llvm::DISubrange *DebugTranslation::translateImpl(DISubrangeAttr attr) {
372 auto getMetadataOrNull = [&](
Attribute attr) -> llvm::Metadata * {
376 llvm::Metadata *metadata =
378 .Case([&](IntegerAttr intAttr) {
380 llvm::Type::getInt64Ty(llvmCtx), intAttr.getInt()));
382 .Case([&](LLVM::DIExpressionAttr expr) {
385 .Case([&](LLVM::DILocalVariableAttr local) {
388 .Case<>([&](LLVM::DIGlobalVariableAttr global) {
391 .Default([&](
Attribute attr) {
return nullptr; });
395 getMetadataOrNull(attr.getLowerBound()),
396 getMetadataOrNull(attr.getUpperBound()),
397 getMetadataOrNull(attr.getStride()));
400 llvm::DICommonBlock *DebugTranslation::translateImpl(DICommonBlockAttr attr) {
403 getMDStringOrNull(attr.getName()),
404 translate(attr.getFile()), attr.getLine());
407 llvm::DIGenericSubrange *
408 DebugTranslation::translateImpl(DIGenericSubrangeAttr attr) {
409 auto getMetadataOrNull = [&](
Attribute attr) -> llvm::Metadata * {
413 llvm::Metadata *metadata =
415 .Case([&](LLVM::DIExpressionAttr expr) {
418 .Case([&](LLVM::DILocalVariableAttr local) {
421 .Case<>([&](LLVM::DIGlobalVariableAttr global) {
424 .Default([&](
Attribute attr) {
return nullptr; });
428 getMetadataOrNull(attr.getCount()),
429 getMetadataOrNull(attr.getLowerBound()),
430 getMetadataOrNull(attr.getUpperBound()),
431 getMetadataOrNull(attr.getStride()));
434 llvm::DISubroutineType *
435 DebugTranslation::translateImpl(DISubroutineTypeAttr attr) {
441 llvmCtx, llvm::DINode::FlagZero, attr.getCallingConvention(),
445 llvm::DIType *DebugTranslation::translateImpl(
DITypeAttr attr) {
453 if (llvm::DINode *node = attrToNode.lookup(attr))
456 llvm::DINode *node =
nullptr;
459 if (
auto recTypeAttr = dyn_cast<DIRecursiveTypeAttrInterface>(attr))
460 if (recTypeAttr.getRecId())
461 node = translateRecursive(recTypeAttr);
465 .Case<DIBasicTypeAttr, DICommonBlockAttr, DICompileUnitAttr,
466 DICompositeTypeAttr, DIDerivedTypeAttr, DIFileAttr,
467 DIGenericSubrangeAttr, DIGlobalVariableAttr,
468 DIImportedEntityAttr, DILabelAttr, DILexicalBlockAttr,
469 DILexicalBlockFileAttr, DILocalVariableAttr, DIModuleAttr,
470 DINamespaceAttr, DINullTypeAttr, DIStringTypeAttr,
471 DISubprogramAttr, DISubrangeAttr, DISubroutineTypeAttr>(
472 [&](
auto attr) {
return translateImpl(attr); });
474 if (node && !node->isTemporary())
475 attrToNode.insert({attr, node});
484 llvm::DILocation *DebugTranslation::translateLoc(
Location loc,
485 llvm::DILocalScope *scope) {
486 if (!debugEmissionIsEnabled)
488 return translateLoc(loc, scope,
nullptr);
492 DebugTranslation::translateExpression(LLVM::DIExpressionAttr attr) {
496 for (
const DIExpressionElemAttr &op : attr.getOperations()) {
497 ops.push_back(op.getOpcode());
498 append_range(ops, op.getArguments());
504 llvm::DIGlobalVariableExpression *
505 DebugTranslation::translateGlobalVariableExpression(
506 LLVM::DIGlobalVariableExpressionAttr attr) {
508 llvmCtx, translate(attr.getVar()), translateExpression(attr.getExpr()));
512 llvm::DILocation *DebugTranslation::translateLoc(
Location loc,
513 llvm::DILocalScope *scope,
514 llvm::DILocation *inlinedAt) {
516 if (isa<UnknownLoc>(loc))
520 auto existingIt = locationToLoc.find(std::make_tuple(loc, scope, inlinedAt));
521 if (existingIt != locationToLoc.end())
522 return existingIt->second;
524 llvm::DILocation *llvmLoc =
nullptr;
525 if (
auto callLoc = dyn_cast<CallSiteLoc>(loc)) {
527 auto *callerLoc = translateLoc(callLoc.getCaller(), scope, inlinedAt);
535 callerLoc = inlinedAt;
537 llvmLoc = translateLoc(callLoc.getCallee(),
nullptr, callerLoc);
542 }
else if (
auto fileLoc = dyn_cast<FileLineColLoc>(loc)) {
548 scope,
const_cast<llvm::DILocation *
>(inlinedAt));
550 }
else if (
auto fusedLoc = dyn_cast<FusedLoc>(loc)) {
554 if (
auto scopedAttr =
555 dyn_cast_or_null<LLVM::DILocalScopeAttr>(fusedLoc.getMetadata()))
556 scope = translate(scopedAttr);
559 llvmLoc = translateLoc(locations.front(), scope, inlinedAt);
560 for (
Location locIt : locations.drop_front()) {
561 llvmLoc = llvm::DILocation::getMergedLocation(
562 llvmLoc, translateLoc(locIt, scope, inlinedAt));
565 }
else if (
auto nameLoc = dyn_cast<NameLoc>(loc)) {
566 llvmLoc = translateLoc(nameLoc.getChildLoc(), scope, inlinedAt);
568 }
else if (
auto opaqueLoc = dyn_cast<OpaqueLoc>(loc)) {
569 llvmLoc = translateLoc(opaqueLoc.getFallbackLocation(), scope, inlinedAt);
571 llvm_unreachable(
"unknown location kind");
574 locationToLoc.try_emplace(std::make_tuple(loc, scope, inlinedAt), llvmLoc);
579 llvm::DIFile *DebugTranslation::translateFile(StringRef fileName) {
580 auto *&file = fileMap[fileName];
585 if (currentWorkingDir.empty())
586 llvm::sys::fs::current_path(currentWorkingDir);
588 StringRef directory = currentWorkingDir;
591 if (llvm::sys::path::is_absolute(fileName)) {
594 auto fileIt = llvm::sys::path::begin(fileName);
595 auto fileE = llvm::sys::path::end(fileName);
596 auto curDirIt = llvm::sys::path::begin(directory);
597 auto curDirE = llvm::sys::path::end(directory);
598 for (; curDirIt != curDirE && *curDirIt == *fileIt; ++curDirIt, ++fileIt)
599 llvm::sys::path::append(dirBuf, *curDirIt);
600 if (std::distance(llvm::sys::path::begin(directory), curDirIt) == 1) {
603 directory = StringRef();
605 for (; fileIt != fileE; ++fileIt)
606 llvm::sys::path::append(fileBuf, *fileIt);
static DINodeT * getDistinctOrUnique(bool isDistinct, Ts &&...args)
Returns a new DINodeT that is either distinct or not, depending on isDistinct.
static WalkResult interruptIfValidLocation(Operation *op)
A utility walker that interrupts if the operation has valid debug information.
static MLIRContext * getContext(OpFoldResult val)
Attributes are known-constant values of operations.
An attribute that associates a referenced attribute with a unique identifier.
This class represents a fused location whose metadata is known to be an instance of the given type.
This class represents a LLVM attribute that describes a local debug info scope.
This class represents the base attribute for all debug info attributes.
This class represents a LLVM attribute that describes a debug info scope.
This class represents a LLVM attribute that describes a debug info type.
This class represents a LLVM attribute that describes a debug info variable.
void finalize()
Finalize the translation of debug information.
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.
DebugTranslation(Operation *module, llvm::Module &llvmModule)
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.
Attribute getDiscardableAttr(StringRef name)
Access a discardable attribute by name, returns an null Attribute if the discardable attribute does n...
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),...
Location getLoc()
The source location the operation was defined or derived from.
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.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...