MLIR
21.0.0git
|
Module import implementation class that provides methods to import globals and functions from an LLVM module into an MLIR module. More...
#include "mlir/Target/LLVMIR/ModuleImport.h"
Public Member Functions | |
ModuleImport (ModuleOp mlirModule, std::unique_ptr< llvm::Module > llvmModule, bool emitExpensiveWarnings, bool importEmptyDICompositeTypes) | |
LogicalResult | initializeImportInterface () |
Calls the LLVMImportInterface initialization that queries the registered dialect interfaces for the supported LLVM IR intrinsics and metadata kinds and builds the dispatch tables. More... | |
LogicalResult | convertFunctions () |
Converts all functions of the LLVM module to MLIR functions. More... | |
LogicalResult | convertComdats () |
Converts all comdat selectors of the LLVM module to MLIR comdat operations. More... | |
LogicalResult | convertGlobals () |
Converts all global variables of the LLVM module to MLIR global variables. More... | |
LogicalResult | convertAliases () |
Converts all aliases of the LLVM module to MLIR variables. More... | |
LogicalResult | convertDataLayout () |
Converts the data layout of the LLVM module to an MLIR data layout specification. More... | |
void | convertTargetTriple () |
Converts target triple of the LLVM module to an MLIR target triple specification. More... | |
void | mapValue (llvm::Value *llvm, Value mlir) |
Stores the mapping between an LLVM value and its MLIR counterpart. More... | |
Value & | mapValue (llvm::Value *value) |
Provides write-once access to store the MLIR value corresponding to the given LLVM value. More... | |
Value | lookupValue (llvm::Value *value) |
Returns the MLIR value mapped to the given LLVM value. More... | |
void | mapNoResultOp (llvm::Instruction *llvm, Operation *mlir) |
Stores a mapping between an LLVM instruction and the imported MLIR operation if the operation returns no result. More... | |
Operation *& | mapNoResultOp (llvm::Instruction *inst) |
Provides write-once access to store the MLIR operation corresponding to the given LLVM instruction if the operation returns no result. More... | |
Operation * | lookupOperation (llvm::Instruction *inst) |
Returns the MLIR operation mapped to the given LLVM instruction. More... | |
void | mapBlock (llvm::BasicBlock *llvm, Block *mlir) |
Stores the mapping between an LLVM block and its MLIR counterpart. More... | |
Block * | lookupBlock (llvm::BasicBlock *block) const |
Returns the MLIR block mapped to the given LLVM block. More... | |
FailureOr< Value > | convertValue (llvm::Value *value) |
Converts an LLVM value to an MLIR value, or returns failure if the conversion fails. More... | |
FailureOr< Value > | convertMetadataValue (llvm::Value *value) |
Converts an LLVM metadata value to an MLIR value, or returns failure if the conversion fails. More... | |
FailureOr< SmallVector< Value > > | convertValues (ArrayRef< llvm::Value * > values) |
Converts a range of LLVM values to a range of MLIR values using the convertValue method, or returns failure if the conversion fails. More... | |
IntegerAttr | matchIntegerAttr (llvm::Value *value) |
Converts value to an integer attribute. Asserts if the matching fails. More... | |
FloatAttr | matchFloatAttr (llvm::Value *value) |
Converts value to a float attribute. Asserts if the matching fails. More... | |
DILocalVariableAttr | matchLocalVariableAttr (llvm::Value *value) |
Converts value to a local variable attribute. More... | |
DILabelAttr | matchLabelAttr (llvm::Value *value) |
Converts value to a label attribute. Asserts if the matching fails. More... | |
FPExceptionBehaviorAttr | matchFPExceptionBehaviorAttr (llvm::Value *value) |
Converts value to a FP exception behavior attribute. More... | |
RoundingModeAttr | matchRoundingModeAttr (llvm::Value *value) |
Converts value to a rounding mode attribute. More... | |
FailureOr< SmallVector< AliasScopeAttr > > | matchAliasScopeAttrs (llvm::Value *value) |
Converts value to an array of alias scopes or returns failure if the conversion fails. More... | |
Location | translateLoc (llvm::DILocation *loc) |
Translates the debug location. More... | |
Type | convertType (llvm::Type *type) |
Converts the type from LLVM to MLIR LLVM dialect. More... | |
LogicalResult | processFunction (llvm::Function *func) |
Imports func into the current module. More... | |
void | processFunctionAttributes (llvm::Function *func, LLVMFuncOp funcOp) |
Converts function attributes of LLVM Function func into LLVM dialect attributes of LLVMFuncOp funcOp . More... | |
void | setIntegerOverflowFlags (llvm::Instruction *inst, Operation *op) const |
Sets the integer overflow flags (nsw/nuw) attribute for the imported operation op given the original instruction inst . More... | |
void | setExactFlag (llvm::Instruction *inst, Operation *op) const |
Sets the exact flag attribute for the imported operation op given the original instruction inst . More... | |
void | setDisjointFlag (llvm::Instruction *inst, Operation *op) const |
Sets the disjoint flag attribute for the imported operation op given the original instruction inst . More... | |
void | setNonNegFlag (llvm::Instruction *inst, Operation *op) const |
Sets the nneg flag attribute for the imported operation op given the original instruction inst . More... | |
void | setFastmathFlagsAttr (llvm::Instruction *inst, Operation *op) const |
Sets the fastmath flags attribute for the imported operation op given the original instruction inst . More... | |
LogicalResult | convertLinkerOptionsMetadata () |
Converts !llvm.linker.options metadata to the llvm.linker.options LLVM dialect operation. More... | |
LogicalResult | convertIdentMetadata () |
Converts !llvm.ident metadata to the llvm.ident LLVM ModuleOp attribute. More... | |
LogicalResult | convertCommandlineMetadata () |
Converts !llvm.commandline metadata to the llvm.commandline LLVM ModuleOp attribute. More... | |
LogicalResult | convertMetadata () |
Converts all LLVM metadata nodes that translate to attributes such as alias analysis or access group metadata, and builds a map from the metadata nodes to the converted attributes. More... | |
Attribute | lookupTBAAAttr (const llvm::MDNode *node) const |
Returns the MLIR attribute mapped to the given LLVM TBAA metadata node . More... | |
FailureOr< SmallVector< AccessGroupAttr > > | lookupAccessGroupAttrs (const llvm::MDNode *node) const |
Returns the access group attributes that map to the access group nodes starting from the access group metadata node . More... | |
LoopAnnotationAttr | translateLoopAnnotationAttr (const llvm::MDNode *node, Location loc) const |
Returns the loop annotation attribute that corresponds to the given LLVM loop metadata node . More... | |
FailureOr< SmallVector< AliasScopeAttr > > | lookupAliasScopeAttrs (const llvm::MDNode *node) const |
Returns the alias scope attributes that map to the alias scope nodes starting from the metadata node . More... | |
void | addDebugIntrinsic (llvm::CallInst *intrinsic) |
Adds a debug intrinsics to the list of intrinsics that should be converted after the function conversion has finished. More... | |
LogicalResult | convertIntrinsicArguments (ArrayRef< llvm::Value * > values, ArrayRef< llvm::OperandBundleUse > opBundles, bool requiresOpBundles, ArrayRef< unsigned > immArgPositions, ArrayRef< StringLiteral > immArgAttrNames, SmallVectorImpl< Value > &valuesOut, SmallVectorImpl< NamedAttribute > &attrsOut) |
Converts the LLVM values for an intrinsic to mixed MLIR values and attributes for LLVM_IntrOpBase. More... | |
void | convertParameterAttributes (llvm::CallBase *call, ArrayAttr &argsAttr, ArrayAttr &resAttr, OpBuilder &builder) |
Converts the parameter and result attributes in argsAttr and resAttr and add them to the callOp . More... | |
Module import implementation class that provides methods to import globals and functions from an LLVM module into an MLIR module.
It holds mappings between the original and translated globals, basic blocks, and values used during the translation. Additionally, it keeps track of the current constant insertion point since LLVM immediate values translate to MLIR operations that are introduced at the beginning of the region.
Definition at line 47 of file ModuleImport.h.
ModuleImport::ModuleImport | ( | ModuleOp | mlirModule, |
std::unique_ptr< llvm::Module > | llvmModule, | ||
bool | emitExpensiveWarnings, | ||
bool | importEmptyDICompositeTypes | ||
) |
Definition at line 162 of file ModuleImport.cpp.
References mlir::OpBuilder::setInsertionPointToStart().
void ModuleImport::addDebugIntrinsic | ( | llvm::CallInst * | intrinsic | ) |
Adds a debug intrinsics to the list of intrinsics that should be converted after the function conversion has finished.
Definition at line 516 of file ModuleImport.cpp.
LogicalResult ModuleImport::convertAliases | ( | ) |
Converts all aliases of the LLVM module to MLIR variables.
Definition at line 650 of file ModuleImport.cpp.
References diag(), mlir::emitError(), and mlir::get().
Referenced by mlir::translateLLVMIRToModule().
LogicalResult ModuleImport::convertComdats | ( | ) |
Converts all comdat selectors of the LLVM module to MLIR comdat operations.
Definition at line 622 of file ModuleImport.cpp.
Referenced by mlir::translateLLVMIRToModule().
LogicalResult ModuleImport::convertCommandlineMetadata | ( | ) |
Converts !llvm.commandline metadata to the llvm.commandline LLVM ModuleOp attribute.
Definition at line 554 of file ModuleImport.cpp.
References mlir::Builder::getStringAttr().
Referenced by convertMetadata().
LogicalResult ModuleImport::convertDataLayout | ( | ) |
Converts the data layout of the LLVM module to an MLIR data layout specification.
Definition at line 660 of file ModuleImport.cpp.
References mlir::emitError(), mlir::emitWarning(), mlir::LLVM::detail::DataLayoutImporter::getDataLayout(), mlir::LLVM::detail::DataLayoutImporter::getLastToken(), and mlir::LLVM::detail::DataLayoutImporter::getUnhandledTokens().
Referenced by mlir::translateLLVMIRToModule().
LogicalResult ModuleImport::convertFunctions | ( | ) |
Converts all functions of the LLVM module to MLIR functions.
Definition at line 681 of file ModuleImport.cpp.
References processFunction().
Referenced by mlir::translateLLVMIRToModule().
LogicalResult ModuleImport::convertGlobals | ( | ) |
Converts all global variables of the LLVM module to MLIR global variables.
Definition at line 632 of file ModuleImport.cpp.
References diag(), mlir::emitError(), mlir::get(), getGlobalCtorsVarName(), and getGlobalDtorsVarName().
Referenced by mlir::translateLLVMIRToModule().
LogicalResult ModuleImport::convertIdentMetadata | ( | ) |
Converts !llvm.ident metadata to the llvm.ident LLVM ModuleOp attribute.
Definition at line 537 of file ModuleImport.cpp.
References mlir::Builder::getStringAttr().
Referenced by convertMetadata().
LogicalResult ModuleImport::convertIntrinsicArguments | ( | ArrayRef< llvm::Value * > | values, |
ArrayRef< llvm::OperandBundleUse > | opBundles, | ||
bool | requiresOpBundles, | ||
ArrayRef< unsigned > | immArgPositions, | ||
ArrayRef< StringLiteral > | immArgAttrNames, | ||
SmallVectorImpl< Value > & | valuesOut, | ||
SmallVectorImpl< NamedAttribute > & | attrsOut | ||
) |
Converts the LLVM values for an intrinsic to mixed MLIR values and attributes for LLVM_IntrOpBase.
Attributes correspond to LLVM immargs. The list immArgPositions
contains the positions of immargs on the LLVM intrinsic, and immArgAttrNames
list (of the same length) contains the corresponding MLIR attribute names.
Definition at line 1408 of file ModuleImport.cpp.
References convertValue(), mlir::detail::DenseArrayAttrImpl< int32_t >::get(), mlir::get(), and getScalarConstantAsAttr().
LogicalResult ModuleImport::convertLinkerOptionsMetadata | ( | ) |
Converts !llvm.linker.options metadata to the llvm.linker.options LLVM dialect operation.
Definition at line 520 of file ModuleImport.cpp.
References mlir::OpBuilder::create(), mlir::Builder::getStrArrayAttr(), and options.
Referenced by convertMetadata().
LogicalResult ModuleImport::convertMetadata | ( | ) |
Converts all LLVM metadata nodes that translate to attributes such as alias analysis or access group metadata, and builds a map from the metadata nodes to the converted attributes.
Returns success if all conversions succeed and failure otherwise.
Definition at line 571 of file ModuleImport.cpp.
References convertCommandlineMetadata(), convertIdentMetadata(), convertLinkerOptionsMetadata(), and mlir::OpBuilder::setInsertionPointToEnd().
Referenced by mlir::translateLLVMIRToModule().
FailureOr< Value > ModuleImport::convertMetadataValue | ( | llvm::Value * | value | ) |
Converts an LLVM metadata value to an MLIR value, or returns failure if the conversion fails.
Uses the convertConstant
method to translate constant LLVM values.
Definition at line 1373 of file ModuleImport.cpp.
void ModuleImport::convertParameterAttributes | ( | llvm::CallBase * | call, |
ArrayAttr & | argsAttr, | ||
ArrayAttr & | resAttr, | ||
OpBuilder & | builder | ||
) |
Converts the parameter and result attributes in argsAttr
and resAttr
and add them to the callOp
.
Definition at line 2211 of file ModuleImport.cpp.
References mlir::Builder::getArrayAttr(), and mlir::Builder::getDictionaryAttr().
Referenced by processFunction().
void ModuleImport::convertTargetTriple | ( | ) |
Converts target triple of the LLVM module to an MLIR target triple specification.
Definition at line 675 of file ModuleImport.cpp.
References mlir::Builder::getStringAttr().
Referenced by mlir::translateLLVMIRToModule().
|
inline |
Converts the type from LLVM to MLIR LLVM dialect.
Definition at line 181 of file ModuleImport.h.
References mlir::LLVM::TypeFromLLVMIRTranslator::translateType().
Referenced by convertVecTypeHint(), and processFunction().
FailureOr< Value > ModuleImport::convertValue | ( | llvm::Value * | value | ) |
Converts an LLVM value to an MLIR value, or returns failure if the conversion fails.
Uses the convertConstant
method to translate constant LLVM values.
Definition at line 1354 of file ModuleImport.cpp.
References diag(), mlir::emitError(), mlir::get(), and translateLoc().
Referenced by convertIntrinsicArguments(), convertValues(), matchFloatAttr(), and matchIntegerAttr().
FailureOr< SmallVector< Value > > ModuleImport::convertValues | ( | ArrayRef< llvm::Value * > | values | ) |
Converts a range of LLVM values to a range of MLIR values using the convertValue
method, or returns failure if the conversion fails.
Definition at line 1396 of file ModuleImport.cpp.
References convertValue().
|
inline |
Calls the LLVMImportInterface initialization that queries the registered dialect interfaces for the supported LLVM IR intrinsics and metadata kinds and builds the dispatch tables.
Returns failure if multiple dialect interfaces translate the same LLVM IR intrinsic.
Definition at line 56 of file ModuleImport.h.
References mlir::LLVMImportInterface::initializeImport().
Referenced by mlir::translateLLVMIRToModule().
FailureOr< SmallVector< AccessGroupAttr > > ModuleImport::lookupAccessGroupAttrs | ( | const llvm::MDNode * | node | ) | const |
Returns the access group attributes that map to the access group nodes starting from the access group metadata node
.
Returns failure, if any of the attributes cannot be found.
Definition at line 2520 of file ModuleImport.cpp.
Referenced by setAccessGroupsAttr().
FailureOr< SmallVector< AliasScopeAttr > > ModuleImport::lookupAliasScopeAttrs | ( | const llvm::MDNode * | node | ) | const |
Returns the alias scope attributes that map to the alias scope nodes starting from the metadata node
.
Returns failure, if any of the attributes cannot be found.
Definition at line 502 of file ModuleImport.cpp.
Referenced by matchAliasScopeAttrs(), setAliasScopesAttr(), and setNoaliasScopesAttr().
|
inline |
Returns the MLIR block mapped to the given LLVM block.
Definition at line 133 of file ModuleImport.h.
Referenced by processFunction().
|
inline |
Returns the MLIR operation mapped to the given LLVM instruction.
Queries valueMapping and noResultOpMapping to support operations with and without result.
Definition at line 119 of file ModuleImport.h.
References lookupValue().
|
inline |
Returns the MLIR attribute mapped to the given LLVM TBAA metadata node
.
Definition at line 236 of file ModuleImport.h.
Referenced by setTBAAAttr().
|
inline |
Returns the MLIR value mapped to the given LLVM value.
Definition at line 94 of file ModuleImport.h.
Referenced by lookupOperation().
|
inline |
Stores the mapping between an LLVM block and its MLIR counterpart.
Definition at line 126 of file ModuleImport.h.
Referenced by processFunction().
|
inline |
Provides write-once access to store the MLIR operation corresponding to the given LLVM instruction if the operation returns no result.
Asserts if the operation returns a result and should be added to valueMapping instead.
Definition at line 107 of file ModuleImport.h.
|
inline |
Stores a mapping between an LLVM instruction and the imported MLIR operation if the operation returns no result.
Asserts if the operation returns a result and should be added to valueMapping instead.
Definition at line 99 of file ModuleImport.h.
|
inline |
Stores the mapping between an LLVM value and its MLIR counterpart.
Definition at line 82 of file ModuleImport.h.
References mapValue().
Referenced by mapValue(), and processFunction().
|
inline |
Provides write-once access to store the MLIR value corresponding to the given LLVM value.
Definition at line 86 of file ModuleImport.h.
FailureOr< SmallVector< AliasScopeAttr > > ModuleImport::matchAliasScopeAttrs | ( | llvm::Value * | value | ) |
Converts value
to an array of alias scopes or returns failure if the conversion fails.
Definition at line 1526 of file ModuleImport.cpp.
References lookupAliasScopeAttrs().
FloatAttr ModuleImport::matchFloatAttr | ( | llvm::Value * | value | ) |
Converts value
to a float attribute. Asserts if the matching fails.
Definition at line 1482 of file ModuleImport.cpp.
References convertValue(), mlir::m_Constant(), and mlir::matchPattern().
FPExceptionBehaviorAttr ModuleImport::matchFPExceptionBehaviorAttr | ( | llvm::Value * | value | ) |
Converts value
to a FP exception behavior attribute.
Asserts if the matching fails.
Definition at line 1505 of file ModuleImport.cpp.
References mlir::Builder::getAttr().
IntegerAttr ModuleImport::matchIntegerAttr | ( | llvm::Value * | value | ) |
Converts value
to an integer attribute. Asserts if the matching fails.
Definition at line 1472 of file ModuleImport.cpp.
References convertValue(), mlir::m_Constant(), and mlir::matchPattern().
DILabelAttr ModuleImport::matchLabelAttr | ( | llvm::Value * | value | ) |
Converts value
to a label attribute. Asserts if the matching fails.
Definition at line 1498 of file ModuleImport.cpp.
DILocalVariableAttr ModuleImport::matchLocalVariableAttr | ( | llvm::Value * | value | ) |
Converts value
to a local variable attribute.
Asserts if the matching fails.
Definition at line 1492 of file ModuleImport.cpp.
RoundingModeAttr ModuleImport::matchRoundingModeAttr | ( | llvm::Value * | value | ) |
Converts value
to a rounding mode attribute.
Asserts if the matching fails.
Definition at line 1515 of file ModuleImport.cpp.
References mlir::Builder::getAttr().
LogicalResult ModuleImport::processFunction | ( | llvm::Function * | func | ) |
Imports func
into the current module.
Definition at line 2292 of file ModuleImport.cpp.
References convertParameterAttributes(), convertType(), mlir::OpBuilder::create(), mlir::OpBuilder::createBlock(), diag(), diagMD(), mlir::emitWarning(), mlir::Region::end(), mlir::detail::enumerate(), getTopologicallySortedBlocks(), mlir::LLVMImportInterface::isConvertibleIntrinsic(), mlir::LLVMImportInterface::isConvertibleMetadata(), kind, lookupBlock(), mapBlock(), mapValue(), processFunctionAttributes(), mlir::OpBuilder::setInsertionPointToEnd(), and mlir::LLVMImportInterface::setMetadataAttrs().
Referenced by convertFunctions().
void ModuleImport::processFunctionAttributes | ( | llvm::Function * | func, |
LLVMFuncOp | funcOp | ||
) |
Converts function attributes of LLVM Function func
into LLVM dialect attributes of LLVMFuncOp funcOp
.
Definition at line 2052 of file ModuleImport.cpp.
References mlir::get(), processMemoryEffects(), and processPassthroughAttrs().
Referenced by processFunction().
void ModuleImport::setDisjointFlag | ( | llvm::Instruction * | inst, |
Operation * | op | ||
) | const |
Sets the disjoint flag attribute for the imported operation op
given the original instruction inst
.
Asserts if the operation does not implement the disjoint flag interface.
Definition at line 724 of file ModuleImport.cpp.
void ModuleImport::setExactFlag | ( | llvm::Instruction * | inst, |
Operation * | op | ||
) | const |
Sets the exact flag attribute for the imported operation op
given the original instruction inst
.
Asserts if the operation does not implement the exact flag interface.
Definition at line 718 of file ModuleImport.cpp.
void ModuleImport::setFastmathFlagsAttr | ( | llvm::Instruction * | inst, |
Operation * | op | ||
) | const |
Sets the fastmath flags attribute for the imported operation op
given the original instruction inst
.
Asserts if the operation does not implement the fastmath interface.
Definition at line 738 of file ModuleImport.cpp.
References contract(), mlir::get(), and mlir::Builder::getContext().
void ModuleImport::setIntegerOverflowFlags | ( | llvm::Instruction * | inst, |
Operation * | op | ||
) | const |
Sets the integer overflow flags (nsw/nuw) attribute for the imported operation op
given the original instruction inst
.
Asserts if the operation does not implement the integer overflow flag interface.
Definition at line 706 of file ModuleImport.cpp.
void ModuleImport::setNonNegFlag | ( | llvm::Instruction * | inst, |
Operation * | op | ||
) | const |
Sets the nneg flag attribute for the imported operation op
given the original instruction inst
.
Asserts if the operation does not implement the nneg flag interface.
Definition at line 732 of file ModuleImport.cpp.
Location ModuleImport::translateLoc | ( | llvm::DILocation * | loc | ) |
Translates the debug location.
Definition at line 1532 of file ModuleImport.cpp.
Referenced by convertValue().
LoopAnnotationAttr ModuleImport::translateLoopAnnotationAttr | ( | const llvm::MDNode * | node, |
Location | loc | ||
) | const |
Returns the loop annotation attribute that corresponds to the given LLVM loop metadata node
.
Definition at line 2525 of file ModuleImport.cpp.
Referenced by setLoopAttr().