Implementation class for module translation. More...

#include "mlir/Target/LLVMIR/ModuleTranslation.h"

Classes
struct	SaveStack
	RAII object calling stackPush/stackPop on construction/destruction. More...

class	StackFrame
	Common CRTP base class for ModuleTranslation stack frames. More...

class	StackFrameBase
	Concrete CRTP base class for ModuleTranslation stack frames. 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...

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	forgetMapping (Region &region)
	Removes the mapping for blocks contained in the region and values defined in these blocks. More...

llvm::MDNode *	getAccessGroup (Operation &opInst, SymbolRefAttr accessGroupRef) const
	Returns the LLVM metadata corresponding to a reference to an mlir LLVM dialect access group operation. More...

llvm::MDNode *	getAliasScope (Operation &opInst, SymbolRefAttr aliasScopeRef) const
	Returns the LLVM metadata corresponding to a reference to an mlir LLVM dialect alias scope operation. More...

llvm::MDNode *	lookupLoopOptionsMetadata (Attribute options) const
	Returns the LLVM metadata corresponding to a llvm loop's codegen options attribute. More...

void	mapLoopOptionsMetadata (Attribute options, llvm::MDNode *metadata)

void	setAccessGroupsMetadata (Operation op, llvm::Instruction inst)

void	setAliasScopeMetadata (Operation op, llvm::Instruction inst)

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::OpenMPIRBuilder *	getOpenMPBuilder ()
	Returns the OpenMP IR builder associated with the LLVM IR module being constructed. More...

const llvm::DILocation *	translateLoc (Location loc, llvm::DILocalScope *scope)
	Translates the given location. More...

LogicalResult	convertBlock (Block &bb, bool ignoreArguments, llvm::IRBuilderBase &builder)
	Translates the contents of the given block to LLVM IR using this translator. 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(const T &)> callback) const
	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)

Detailed Description

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 52 of file ModuleTranslation.h.

Member Function Documentation

◆ convertBlock()

LogicalResult ModuleTranslation::convertBlock	(	Block &	bb,
		bool	ignoreArguments,
		llvm::IRBuilderBase &	builder
	)

Translates the contents of the given block to LLVM IR using this translator.

Convert block to LLVM IR.

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.

Unless ignoreArguments is set, emit PHI nodes to define values corresponding to the MLIR block arguments. These nodes are not connected to the source basic blocks, which may not exist yet. Uses builder to construct the LLVM IR. Expects the LLVM IR basic block to have been created for bb and included in the block mapping. Inserts new instructions at the end of the block and leaves builder in a state suitable for further insertion into the end of the block.

Definition at line 580 of file ModuleTranslation.cpp.

References mlir::emitError(), mlir::failed(), mlir::failure(), mlir::Block::front(), mlir::Block::getArguments(), mlir::Operation::getLoc(), mlir::Block::getPredecessors(), mlir::LLVM::isCompatibleType(), and mlir::success().

Referenced by convertDataOp(), convertOmpAtomicCapture(), convertOmpOpRegions(), and inlineConvertOmpRegions().

◆ convertType()

llvm::Type * ModuleTranslation::convertType ( Type type )

Converts the type from MLIR LLVM dialect to LLVM.

Definition at line 1117 of file ModuleTranslation.cpp.

Referenced by convertOmpAtomicCapture(), convertOmpAtomicRead(), convertOmpAtomicUpdate(), convertOmpAtomicWrite(), convertOmpOpRegions(), convertOmpReductionOp(), convertOmpWsLoop(), and convertOperationImpl().

◆ forgetMapping()

void ModuleTranslation::forgetMapping ( Region & region )

Removes the mapping for blocks contained in the region and values defined in these blocks.

Definition at line 421 of file ModuleTranslation.cpp.

References value.

Referenced by convertOmpOpRegions(), and inlineConvertOmpRegions().

◆ getAccessGroup()

llvm::MDNode * ModuleTranslation::getAccessGroup	(	Operation &	opInst,
		SymbolRefAttr	accessGroupRef
	)		const

Returns the LLVM metadata corresponding to a reference to an mlir LLVM dialect access group operation.

Definition at line 1006 of file ModuleTranslation.cpp.

References mlir::Operation::getParentOp(), mlir::SymbolTable::lookupNearestSymbolFrom(), and mlir::success().

Referenced by setLoopMetadata().

◆ getAliasScope()

llvm::MDNode * ModuleTranslation::getAliasScope	(	Operation &	opInst,
		SymbolRefAttr	aliasScopeRef
	)		const

Returns the LLVM metadata corresponding to a reference to an mlir LLVM dialect alias scope operation.

Definition at line 1087 of file ModuleTranslation.cpp.

References mlir::Operation::getParentOp(), and mlir::SymbolTable::lookupNearestSymbolFrom().

◆ getContext()

MLIRContext& mlir::LLVM::ModuleTranslation::getContext ( )

inline

Returns the MLIR context of the module being translated.

Definition at line 154 of file ModuleTranslation.h.

Referenced by convertOperationImpl().

◆ getLLVMContext()

llvm::LLVMContext& mlir::LLVM::ModuleTranslation::getLLVMContext ( ) const

inline

Returns the LLVM context in which the IR is being constructed.

Definition at line 157 of file ModuleTranslation.h.

Referenced by convertOmpCritical(), convertOperationImpl(), getLdMatrixIntrinsicId(), and mlir::LLVM::detail::getLLVMConstant().

◆ getOpenMPBuilder()

llvm::OpenMPIRBuilder* mlir::LLVM::ModuleTranslation::getOpenMPBuilder ( )

inline

Returns the OpenMP IR builder associated with the LLVM IR module being constructed.

Definition at line 167 of file ModuleTranslation.h.

Referenced by convertDataOp(), convertOmpAtomicCapture(), convertOmpAtomicRead(), convertOmpAtomicUpdate(), convertOmpAtomicWrite(), convertOmpCritical(), convertOmpMaster(), convertOmpOrdered(), convertOmpOrderedRegion(), convertOmpParallel(), convertOmpSections(), convertOmpSimdLoop(), convertOmpSingle(), convertOmpTaskgroupOp(), convertOmpTaskOp(), convertOmpThreadprivate(), convertOmpWsLoop(), convertStandaloneDataOp(), and processOperands().

◆ getOrInsertNamedModuleMetadata()

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 1136 of file ModuleTranslation.cpp.

Referenced by getLdMatrixIntrinsicId().

◆ lookupBlock()

llvm::BasicBlock* mlir::LLVM::ModuleTranslation::lookupBlock ( Block * block ) const

inline

Finds an LLVM IR basic block that corresponds to the given MLIR block.

Definition at line 98 of file ModuleTranslation.h.

Referenced by mlir::LLVM::detail::connectPHINodes(), convertDataOp(), convertOmpOpRegions(), and convertOperationImpl().

◆ lookupBranch()

llvm::Instruction* mlir::LLVM::ModuleTranslation::lookupBranch ( Operation * op ) const

inline

Finds an LLVM IR instruction that corresponds to the given MLIR operation with successors.

Definition at line 113 of file ModuleTranslation.h.

Referenced by mlir::LLVM::detail::connectPHINodes().

◆ lookupFunction()

llvm::Function* mlir::LLVM::ModuleTranslation::lookupFunction ( StringRef name ) const

inline

Finds an LLVM IR function by its name.

Definition at line 66 of file ModuleTranslation.h.

Referenced by convertDataOp(), convertOperationImpl(), convertStandaloneDataOp(), createDeviceFunctionCall(), getLdMatrixIntrinsicId(), and mlir::LLVM::detail::getLLVMConstant().

◆ lookupGlobal()

llvm::GlobalValue* mlir::LLVM::ModuleTranslation::lookupGlobal ( Operation * op )

inline

Finds an LLVM IR global value that corresponds to the given MLIR operation defining a global value.

Definition at line 161 of file ModuleTranslation.h.

Referenced by convertOmpThreadprivate(), and convertOperationImpl().

◆ lookupLoopOptionsMetadata()

llvm::MDNode* mlir::LLVM::ModuleTranslation::lookupLoopOptionsMetadata ( Attribute options ) const

inline

Returns the LLVM metadata corresponding to a llvm loop's codegen options attribute.

Definition at line 133 of file ModuleTranslation.h.

Referenced by setLoopMetadata().

◆ lookupValue()

llvm::Value* mlir::LLVM::ModuleTranslation::lookupValue ( Value value ) const

inline

Finds an LLVM IR value corresponding to the given MLIR value.

Definition at line 83 of file ModuleTranslation.h.

Referenced by mlir::LLVM::detail::connectPHINodes(), convertOmpAtomicCapture(), convertOmpAtomicRead(), convertOmpAtomicUpdate(), convertOmpAtomicWrite(), convertOmpOpRegions(), convertOmpParallel(), convertOmpReductionOp(), convertOmpSimdLoop(), convertOmpWsLoop(), convertOperationImpl(), and processOperands().

◆ 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 1122 of file ModuleTranslation.cpp.

Referenced by convertOmpOrdered(), convertOperationImpl(), and inlineConvertOmpRegions().

◆ mapBlock()

void mlir::LLVM::ModuleTranslation::mapBlock	(	Block *	mlir,
		llvm::BasicBlock *	llvm
	)

inline

Stores the mapping between an MLIR block and LLVM IR basic block.

Definition at line 91 of file ModuleTranslation.h.

Referenced by convertDataOp(), convertOmpAtomicCapture(), convertOmpOpRegions(), and inlineConvertOmpRegions().

◆ mapBranch()

void mlir::LLVM::ModuleTranslation::mapBranch	(	Operation *	mlir,
		llvm::Instruction *	llvm
	)

inline

Stores the mapping between an MLIR operation with successors and a corresponding LLVM IR instruction.

Definition at line 104 of file ModuleTranslation.h.

Referenced by convertOperationImpl().

◆ mapFunction()

void mlir::LLVM::ModuleTranslation::mapFunction	(	StringRef	name,
		llvm::Function *	func
	)

inline

Stores the mapping between a function name and its LLVM IR representation.

Definition at line 58 of file ModuleTranslation.h.

◆ mapLoopOptionsMetadata()

void mlir::LLVM::ModuleTranslation::mapLoopOptionsMetadata	(	Attribute	options,
		llvm::MDNode *	metadata
	)

inline

Definition at line 137 of file ModuleTranslation.h.

Referenced by setLoopMetadata().

◆ mapValue() [1/2]

void mlir::LLVM::ModuleTranslation::mapValue	(	Value	mlir,
		llvm::Value *	llvm
	)

inline

Stores the mapping between an MLIR value and its LLVM IR counterpart.

Definition at line 71 of file ModuleTranslation.h.

References mapValue().

Referenced by convertOmpAtomicCapture(), convertOmpReductionOp(), convertOmpSimdLoop(), convertOmpThreadprivate(), convertOmpWsLoop(), convertOperationImpl(), and mapValue().

◆ mapValue() [2/2]

llvm::Value*& mlir::LLVM::ModuleTranslation::mapValue ( Value value )

inline

Provides write-once access to store the LLVM IR value corresponding to the given MLIR value.

Definition at line 75 of file ModuleTranslation.h.

References value.

◆ setAccessGroupsMetadata()

void ModuleTranslation::setAccessGroupsMetadata	(	Operation *	op,
		llvm::Instruction *	inst
	)

Definition at line 1028 of file ModuleTranslation.cpp.

References mlir::Operation::getAttrOfType(), mlir::SymbolTable::lookupNearestSymbolFrom(), and mlir::success().

◆ setAliasScopeMetadata()

void ModuleTranslation::setAliasScopeMetadata	(	Operation *	op,
		llvm::Instruction *	inst
	)

Definition at line 1098 of file ModuleTranslation.cpp.

References mlir::Operation::getAttrOfType().

◆ stackPop()

void mlir::LLVM::ModuleTranslation::stackPop ( )

inline

Pops the last element from the ModuleTranslation stack.

Definition at line 233 of file ModuleTranslation.h.

◆ stackPush()

template<typename T , typename... Args>

void mlir::LLVM::ModuleTranslation::stackPush ( Args &&... args )

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 225 of file ModuleTranslation.h.

References value.

◆ stackWalk()

template<typename T >

WalkResult mlir::LLVM::ModuleTranslation::stackWalk ( llvm::function_ref< WalkResult(const T &)> callback ) const

inline

Calls callback for every ModuleTranslation stack frame of type T starting from the top of the stack.

Definition at line 239 of file ModuleTranslation.h.

References value, and mlir::WalkResult::wasInterrupted().

Referenced by convertOmpReductionOp(), and findAllocaInsertPoint().

◆ symbolTable()

SymbolTableCollection& mlir::LLVM::ModuleTranslation::symbolTable ( )

inline

Definition at line 268 of file ModuleTranslation.h.

References mlir::LLVM::detail::connectPHINodes(), mlir::LLVM::detail::createIntrinsicCall(), mlir::LLVM::detail::getLLVMConstant(), and mlir::LLVM::detail::getTopologicallySortedBlocks().

Referenced by convertOmpThreadprivate(), and convertOperationImpl().

◆ translateLoc()

const llvm::DILocation * ModuleTranslation::translateLoc	(	Location	loc,
		llvm::DILocalScope *	scope
	)

Translates the given location.

Definition at line 1131 of file ModuleTranslation.cpp.

Friends And Related Function Documentation

◆ mlir::translateModuleToLLVMIR

std::unique_ptr<llvm::Module> mlir::translateModuleToLLVMIR	(	Operation *	,
		llvm::LLVMContext &	,
		StringRef
	)

friend

The documentation for this class was generated from the following files:

include/mlir/Target/LLVMIR/ModuleTranslation.h
lib/Target/LLVMIR/ModuleTranslation.cpp

Classes

Public Member Functions

Friends

Detailed Description

Member Function Documentation

◆ convertBlock()

◆ convertType()

◆ forgetMapping()

◆ getAccessGroup()

◆ getAliasScope()

◆ getContext()

◆ getLLVMContext()

◆ getOpenMPBuilder()

◆ getOrInsertNamedModuleMetadata()

◆ lookupBlock()

◆ lookupBranch()

◆ lookupFunction()

◆ lookupGlobal()

◆ lookupLoopOptionsMetadata()

◆ lookupValue()

◆ lookupValues()

◆ mapBlock()

◆ mapBranch()

◆ mapFunction()

◆ mapLoopOptionsMetadata()

◆ mapValue() [1/2]

◆ mapValue() [2/2]

◆ setAccessGroupsMetadata()

◆ setAliasScopeMetadata()

◆ stackPop()

◆ stackPush()

◆ stackWalk()

◆ symbolTable()

◆ translateLoc()

Friends And Related Function Documentation

◆ mlir::translateModuleToLLVMIR