MLIR
22.0.0git
|
Implementation class for module translation. More...
#include "mlir/Target/LLVMIR/ModuleTranslation.h"
Public Types | |
template<typename T > | |
using | SaveStack = SaveStateStack< T, ModuleTranslation > |
RAII object calling stackPush/stackPop on construction/destruction. More... | |
Public Member Functions | |
void | mapFunction (StringRef name, llvm::Function *func) |
Stores the mapping between a function name and its LLVM IR representation. More... | |
llvm::Function * | lookupFunction (StringRef name) const |
Finds an LLVM IR function by its name. More... | |
void | mapValue (Value mlir, llvm::Value *llvm) |
Stores the mapping between an MLIR value and its LLVM IR counterpart. More... | |
llvm::Value *& | mapValue (Value value) |
Provides write-once access to store the LLVM IR value corresponding to the given MLIR value. More... | |
llvm::Value * | lookupValue (Value value) const |
Finds an LLVM IR value corresponding to the given MLIR value. More... | |
SmallVector< llvm::Value * > | lookupValues (ValueRange values) |
Looks up remapped a list of remapped values. More... | |
void | mapBlock (Block *mlir, llvm::BasicBlock *llvm) |
Stores the mapping between an MLIR block and LLVM IR basic block. More... | |
llvm::BasicBlock * | lookupBlock (Block *block) const |
Finds an LLVM IR basic block that corresponds to the given MLIR block. More... | |
llvm::CanonicalLoopInfo * | lookupOMPLoop (omp::NewCliOp mlir) const |
Find the LLVM-IR loop that represents an MLIR loop. More... | |
llvm::CanonicalLoopInfo * | lookupOMPLoop (Value mlir) const |
Find the LLVM-IR loop that represents an MLIR loop. More... | |
void | invalidateOmpLoop (omp::NewCliOp mlir) |
Mark an OpenMP loop as having been consumed. More... | |
void | invalidateOmpLoop (Value mlir) |
Mark an OpenMP loop as having been consumed. More... | |
void | mapOmpLoop (omp::NewCliOp mlir, llvm::CanonicalLoopInfo *llvm) |
Map an MLIR OpenMP dialect CanonicalLoopInfo to its lowered LLVM-IR OpenMPIRBuilder CanonicalLoopInfo. More... | |
void | mapOmpLoop (Value mlir, llvm::CanonicalLoopInfo *llvm) |
Map an MLIR OpenMP dialect CanonicalLoopInfo to its lowered LLVM-IR OpenMPIRBuilder CanonicalLoopInfo. More... | |
void | mapBranch (Operation *mlir, llvm::Instruction *llvm) |
Stores the mapping between an MLIR operation with successors and a corresponding LLVM IR instruction. More... | |
llvm::Instruction * | lookupBranch (Operation *op) const |
Finds an LLVM IR instruction that corresponds to the given MLIR operation with successors. More... | |
void | mapCall (Operation *mlir, llvm::CallInst *llvm) |
Stores a mapping between an MLIR call operation and a corresponding LLVM call instruction. More... | |
llvm::CallInst * | lookupCall (Operation *op) const |
Finds an LLVM call instruction that corresponds to the given MLIR call operation. More... | |
void | mapUnresolvedBlockAddress (BlockAddressOp op, llvm::Value *cst) |
Maps a blockaddress operation to its corresponding placeholder LLVM value. More... | |
void | mapBlockAddress (BlockAddressAttr attr, llvm::BasicBlock *block) |
Maps a BlockAddressAttr to its corresponding LLVM basic block. More... | |
llvm::BasicBlock * | lookupBlockAddress (BlockAddressAttr attr) const |
Finds the LLVM basic block that corresponds to the given BlockAddressAttr. More... | |
void | forgetMapping (Region ®ion) |
Removes the mapping for blocks contained in the region and values defined in these blocks. More... | |
llvm::MDNode * | getOrCreateAliasScope (AliasScopeAttr aliasScopeAttr) |
Returns the LLVM metadata corresponding to a mlir LLVM dialect alias scope attribute. More... | |
llvm::MDNode * | getOrCreateAliasScopes (ArrayRef< AliasScopeAttr > aliasScopeAttrs) |
Returns the LLVM metadata corresponding to an array of mlir LLVM dialect alias scope attributes. More... | |
void | setAccessGroupsMetadata (AccessGroupOpInterface op, llvm::Instruction *inst) |
void | setAliasScopeMetadata (AliasAnalysisOpInterface op, llvm::Instruction *inst) |
void | setTBAAMetadata (AliasAnalysisOpInterface op, llvm::Instruction *inst) |
Sets LLVM TBAA metadata for memory operations that have TBAA attributes. More... | |
void | setDereferenceableMetadata (DereferenceableOpInterface op, llvm::Instruction *inst) |
Sets LLVM dereferenceable metadata for operations that have dereferenceable attributes. More... | |
void | setBranchWeightsMetadata (WeightedBranchOpInterface op) |
Sets LLVM profiling metadata for operations that have branch weights. More... | |
void | setLoopMetadata (Operation *op, llvm::Instruction *inst) |
Sets LLVM loop metadata for branch operations that have a loop annotation attribute. More... | |
void | setDisjointFlag (Operation *op, llvm::Value *value) |
Sets the disjoint flag attribute for the exported instruction value given the original operation op . More... | |
llvm::Type * | convertType (Type type) |
Converts the type from MLIR LLVM dialect to LLVM. More... | |
MLIRContext & | getContext () |
Returns the MLIR context of the module being translated. More... | |
llvm::LLVMContext & | getLLVMContext () const |
Returns the LLVM context in which the IR is being constructed. More... | |
llvm::GlobalValue * | lookupGlobal (Operation *op) |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining a global value. More... | |
llvm::GlobalValue * | lookupAlias (Operation *op) |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining a global alias value. More... | |
llvm::GlobalValue * | lookupIFunc (Operation *op) |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining an IFunc. More... | |
llvm::OpenMPIRBuilder * | getOpenMPBuilder () |
Returns the OpenMP IR builder associated with the LLVM IR module being constructed. More... | |
llvm::Module * | getLLVMModule () |
Returns the LLVM module in which the IR is being constructed. More... | |
llvm::DILocation * | translateLoc (Location loc, llvm::DILocalScope *scope) |
Translates the given location. More... | |
llvm::DIExpression * | translateExpression (LLVM::DIExpressionAttr attr) |
Translates the given LLVM DWARF expression metadata. More... | |
llvm::DIGlobalVariableExpression * | translateGlobalVariableExpression (LLVM::DIGlobalVariableExpressionAttr attr) |
Translates the given LLVM global variable expression metadata. More... | |
llvm::Metadata * | translateDebugInfo (LLVM::DINodeAttr attr) |
Translates the given LLVM debug info metadata. More... | |
llvm::RoundingMode | translateRoundingMode (LLVM::RoundingMode rounding) |
Translates the given LLVM rounding mode metadata. More... | |
llvm::fp::ExceptionBehavior | translateFPExceptionBehavior (LLVM::FPExceptionBehavior exceptionBehavior) |
Translates the given LLVM FP exception behavior metadata. More... | |
LogicalResult | convertBlock (Block &bb, bool ignoreArguments, llvm::IRBuilderBase &builder) |
Translates the contents of the given block to LLVM IR using this translator. More... | |
LogicalResult | convertArgAndResultAttrs (ArgAndResultAttrsOpInterface attrsOp, llvm::CallBase *call, ArrayRef< unsigned > immArgPositions={}) |
Converts argument and result attributes from attrsOp to LLVM IR attributes on the call instruction. More... | |
llvm::NamedMDNode * | getOrInsertNamedModuleMetadata (StringRef name) |
Gets the named metadata in the LLVM IR module being constructed, creating it if it does not exist. More... | |
template<typename T , typename... Args> | |
void | stackPush (Args &&...args) |
Creates a stack frame of type T on ModuleTranslation stack. More... | |
void | stackPop () |
Pops the last element from the ModuleTranslation stack. More... | |
template<typename T > | |
WalkResult | stackWalk (llvm::function_ref< WalkResult(T &)> callback) |
Calls callback for every ModuleTranslation stack frame of type T starting from the top of the stack. More... | |
SymbolTableCollection & | symbolTable () |
Friends | |
std::unique_ptr< llvm::Module > | mlir::translateModuleToLLVMIR (Operation *, llvm::LLVMContext &, StringRef, bool) |
Implementation class for module translation.
Holds a reference to the module being translated, and the mappings between the original and the translated functions, basic blocks and values. It is practically easier to hold these mappings in one class since the conversion of control flow operations needs to look up block and function mappings.
Definition at line 64 of file ModuleTranslation.h.
using mlir::LLVM::ModuleTranslation::SaveStack = SaveStateStack<T, ModuleTranslation> |
RAII object calling stackPush/stackPop on construction/destruction.
Definition at line 348 of file ModuleTranslation.h.
LogicalResult ModuleTranslation::convertArgAndResultAttrs | ( | ArgAndResultAttrsOpInterface | attrsOp, |
llvm::CallBase * | call, | ||
ArrayRef< unsigned > | immArgPositions = {} |
||
) |
Converts argument and result attributes from attrsOp
to LLVM IR attributes on the call
instruction.
Returns failure if conversion fails. The immArgPositions
parameter is only relevant for intrinsics. It specifies the positions of immediate arguments, which do not have associated argument attributes in MLIR and should be skipped during attribute mapping.
Definition at line 1766 of file ModuleTranslation.cpp.
References mlir::remark::failed().
|
inline |
Translates the contents of the given block to LLVM IR using this translator.
The LLVM IR basic block corresponding to the given block is expected to exist in the mapping of this translator. Uses builder
to translate the IR, leaving it at the end of the block. If ignoreArguments
is set, does not produce PHI nodes for the block arguments. Otherwise, the PHI nodes are constructed for block arguments but are not connected to the predecessors that may not exist yet.
Definition at line 305 of file ModuleTranslation.h.
llvm::Type * ModuleTranslation::convertType | ( | Type | type | ) |
Converts the type from MLIR LLVM dialect to LLVM.
Definition at line 2233 of file ModuleTranslation.cpp.
References mlir::LLVM::TypeToLLVMIRTranslator::translateType().
Referenced by convertFunctionKernelAttributes(), convertParameterAttr(), and mlir::LLVM::detail::createIntrinsicCall().
void ModuleTranslation::forgetMapping | ( | Region & | region | ) |
Removes the mapping for blocks contained in the region and values defined in these blocks.
Definition at line 776 of file ModuleTranslation.cpp.
|
inline |
Returns the MLIR context of the module being translated.
Definition at line 248 of file ModuleTranslation.h.
References mlir::Operation::getContext().
|
inline |
Returns the LLVM context in which the IR is being constructed.
Definition at line 251 of file ModuleTranslation.h.
Referenced by mlir::LLVM::detail::getLLVMConstant(), getOrCreateAliasScopes(), setBranchWeightsMetadata(), and setDereferenceableMetadata().
|
inline |
Returns the LLVM module in which the IR is being constructed.
Definition at line 276 of file ModuleTranslation.h.
llvm::OpenMPIRBuilder * ModuleTranslation::getOpenMPBuilder | ( | ) |
Returns the OpenMP IR builder associated with the LLVM IR module being constructed.
Definition at line 2246 of file ModuleTranslation.cpp.
llvm::MDNode * ModuleTranslation::getOrCreateAliasScope | ( | AliasScopeAttr | aliasScopeAttr | ) |
Returns the LLVM metadata corresponding to a mlir LLVM dialect alias scope attribute.
Creates the metadata node if it has not been converted before.
Definition at line 1991 of file ModuleTranslation.cpp.
References mlir::get().
Referenced by getOrCreateAliasScopes().
llvm::MDNode * ModuleTranslation::getOrCreateAliasScopes | ( | ArrayRef< AliasScopeAttr > | aliasScopeAttrs | ) |
Returns the LLVM metadata corresponding to an array of mlir LLVM dialect alias scope attributes.
Creates the metadata nodes if they have not been converted before.
Definition at line 2036 of file ModuleTranslation.cpp.
References mlir::get(), getLLVMContext(), and getOrCreateAliasScope().
Referenced by setAliasScopeMetadata().
llvm::NamedMDNode * ModuleTranslation::getOrInsertNamedModuleMetadata | ( | StringRef | name | ) |
Gets the named metadata in the LLVM IR module being constructed, creating it if it does not exist.
Definition at line 2296 of file ModuleTranslation.cpp.
|
inline |
Mark an OpenMP loop as having been consumed.
Definition at line 128 of file ModuleTranslation.h.
Referenced by invalidateOmpLoop().
|
inline |
Mark an OpenMP loop as having been consumed.
Definition at line 131 of file ModuleTranslation.h.
References invalidateOmpLoop().
|
inline |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining a global alias value.
Definition at line 261 of file ModuleTranslation.h.
|
inline |
Finds an LLVM IR basic block that corresponds to the given MLIR block.
Definition at line 111 of file ModuleTranslation.h.
|
inline |
Finds the LLVM basic block that corresponds to the given BlockAddressAttr.
Definition at line 197 of file ModuleTranslation.h.
|
inline |
Finds an LLVM IR instruction that corresponds to the given MLIR operation with successors.
Definition at line 161 of file ModuleTranslation.h.
Referenced by setBranchWeightsMetadata().
|
inline |
Finds an LLVM call instruction that corresponds to the given MLIR call operation.
Definition at line 175 of file ModuleTranslation.h.
Referenced by setBranchWeightsMetadata().
|
inline |
Finds an LLVM IR function by its name.
Definition at line 79 of file ModuleTranslation.h.
Referenced by mlir::LLVM::detail::getLLVMConstant().
|
inline |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining a global value.
Definition at line 255 of file ModuleTranslation.h.
|
inline |
Finds an LLVM IR global value that corresponds to the given MLIR operation defining an IFunc.
Definition at line 267 of file ModuleTranslation.h.
|
inline |
Find the LLVM-IR loop that represents an MLIR loop.
Definition at line 116 of file ModuleTranslation.h.
Referenced by lookupOMPLoop().
|
inline |
Find the LLVM-IR loop that represents an MLIR loop.
Definition at line 123 of file ModuleTranslation.h.
References lookupOMPLoop().
|
inline |
Finds an LLVM IR value corresponding to the given MLIR value.
Definition at line 96 of file ModuleTranslation.h.
Referenced by lookupValues().
SmallVector< llvm::Value * > ModuleTranslation::lookupValues | ( | ValueRange | values | ) |
Looks up remapped a list of remapped values.
A helper to look up remapped operands in the value remapping table.
Definition at line 2238 of file ModuleTranslation.cpp.
References lookupValue().
Referenced by mlir::LLVM::detail::createIntrinsicCall().
|
inline |
Stores the mapping between an MLIR block and LLVM IR basic block.
Definition at line 104 of file ModuleTranslation.h.
|
inline |
Maps a BlockAddressAttr to its corresponding LLVM basic block.
Definition at line 189 of file ModuleTranslation.h.
|
inline |
Stores the mapping between an MLIR operation with successors and a corresponding LLVM IR instruction.
Definition at line 152 of file ModuleTranslation.h.
|
inline |
Stores a mapping between an MLIR call operation and a corresponding LLVM call instruction.
Definition at line 167 of file ModuleTranslation.h.
|
inline |
Stores the mapping between a function name and its LLVM IR representation.
Definition at line 71 of file ModuleTranslation.h.
|
inline |
Map an MLIR OpenMP dialect CanonicalLoopInfo to its lowered LLVM-IR OpenMPIRBuilder CanonicalLoopInfo.
Definition at line 137 of file ModuleTranslation.h.
Referenced by mapOmpLoop().
|
inline |
Map an MLIR OpenMP dialect CanonicalLoopInfo to its lowered LLVM-IR OpenMPIRBuilder CanonicalLoopInfo.
Definition at line 146 of file ModuleTranslation.h.
References mapOmpLoop().
|
inline |
Maps a blockaddress operation to its corresponding placeholder LLVM value.
Definition at line 181 of file ModuleTranslation.h.
|
inline |
Stores the mapping between an MLIR value and its LLVM IR counterpart.
Definition at line 84 of file ModuleTranslation.h.
References mapValue().
Referenced by mapValue().
|
inline |
Provides write-once access to store the LLVM IR value corresponding to the given MLIR value.
Definition at line 88 of file ModuleTranslation.h.
void ModuleTranslation::setAccessGroupsMetadata | ( | AccessGroupOpInterface | op, |
llvm::Instruction * | inst | ||
) |
Definition at line 1984 of file ModuleTranslation.cpp.
void ModuleTranslation::setAliasScopeMetadata | ( | AliasAnalysisOpInterface | op, |
llvm::Instruction * | inst | ||
) |
Definition at line 2045 of file ModuleTranslation.cpp.
References getOrCreateAliasScopes(), and kind.
void ModuleTranslation::setBranchWeightsMetadata | ( | WeightedBranchOpInterface | op | ) |
Sets LLVM profiling metadata for operations that have branch weights.
Definition at line 2101 of file ModuleTranslation.cpp.
References getLLVMContext(), lookupBranch(), and lookupCall().
void ModuleTranslation::setDereferenceableMetadata | ( | DereferenceableOpInterface | op, |
llvm::Instruction * | inst | ||
) |
Sets LLVM dereferenceable metadata for operations that have dereferenceable attributes.
Definition at line 2085 of file ModuleTranslation.cpp.
References mlir::get(), and getLLVMContext().
void ModuleTranslation::setDisjointFlag | ( | Operation * | op, |
llvm::Value * | value | ||
) |
Sets the disjoint flag attribute for the exported instruction value
given the original operation op
.
Asserts if the operation does not implement the disjoint flag interface, and asserts if the value is an instruction that implements the disjoint flag.
Definition at line 2226 of file ModuleTranslation.cpp.
void ModuleTranslation::setLoopMetadata | ( | Operation * | op, |
llvm::Instruction * | inst | ||
) |
Sets LLVM loop metadata for branch operations that have a loop annotation attribute.
Definition at line 2213 of file ModuleTranslation.cpp.
void ModuleTranslation::setTBAAMetadata | ( | AliasAnalysisOpInterface | op, |
llvm::Instruction * | inst | ||
) |
Sets LLVM TBAA metadata for memory operations that have TBAA attributes.
Definition at line 2065 of file ModuleTranslation.cpp.
|
inline |
Pops the last element from the ModuleTranslation stack.
Definition at line 337 of file ModuleTranslation.h.
References mlir::StateStack::stackPop().
|
inline |
Creates a stack frame of type T
on ModuleTranslation stack.
T
must be derived from StackFrameBase<T>
and constructible from the provided arguments. Doing this before entering the region of the op being translated makes the frame available when translating ops within that region.
Definition at line 332 of file ModuleTranslation.h.
References mlir::StateStack::stackPush().
|
inline |
Calls callback
for every ModuleTranslation stack frame of type T
starting from the top of the stack.
Definition at line 342 of file ModuleTranslation.h.
References mlir::StateStack::stackWalk().
|
inline |
Definition at line 350 of file ModuleTranslation.h.
llvm::Metadata * ModuleTranslation::translateDebugInfo | ( | LLVM::DINodeAttr | attr | ) |
Translates the given LLVM debug info metadata.
Definition at line 2281 of file ModuleTranslation.cpp.
llvm::DIExpression * ModuleTranslation::translateExpression | ( | LLVM::DIExpressionAttr | attr | ) |
Translates the given LLVM DWARF expression metadata.
Definition at line 2271 of file ModuleTranslation.cpp.
llvm::fp::ExceptionBehavior ModuleTranslation::translateFPExceptionBehavior | ( | LLVM::FPExceptionBehavior | exceptionBehavior | ) |
Translates the given LLVM FP exception behavior metadata.
Definition at line 2290 of file ModuleTranslation.cpp.
llvm::DIGlobalVariableExpression * ModuleTranslation::translateGlobalVariableExpression | ( | LLVM::DIGlobalVariableExpressionAttr | attr | ) |
Translates the given LLVM global variable expression metadata.
Definition at line 2276 of file ModuleTranslation.cpp.
llvm::DILocation * ModuleTranslation::translateLoc | ( | Location | loc, |
llvm::DILocalScope * | scope | ||
) |
Translates the given location.
Definition at line 2265 of file ModuleTranslation.cpp.
llvm::RoundingMode ModuleTranslation::translateRoundingMode | ( | LLVM::RoundingMode | rounding | ) |
Translates the given LLVM rounding mode metadata.
Definition at line 2286 of file ModuleTranslation.cpp.
|
friend |