14 #ifndef MLIR_TARGET_LLVMIR_MODULEIMPORT_H
15 #define MLIR_TARGET_LLVMIR_MODULEIMPORT_H
26 class DbgVariableIntrinsic;
36 class DataLayoutImporter;
38 class LoopAnnotationImporter;
49 ModuleImport(ModuleOp mlirModule, std::unique_ptr<llvm::Module> llvmModule,
50 bool emitExpensiveWarnings);
79 assert(
mlir ==
nullptr &&
80 "attempting to map a value that is already mapped");
100 assert(inst->getType()->isVoidTy() &&
101 "attempting to map an operation that returns a result");
102 assert(
mlir ==
nullptr &&
103 "attempting to map an operation that is already mapped");
112 return value.getDefiningOp();
113 return noResultOpMapping.lookup(inst);
118 auto result = blockMapping.try_emplace(
llvm,
mlir);
120 assert(result.second &&
"attempting to map a block that is already mapped");
125 return blockMapping.lookup(block);
193 return tbaaMapping.lookup(node);
231 void clearRegionState() {
232 valueMapping.clear();
233 noResultOpMapping.clear();
234 blockMapping.clear();
235 debugIntrinsics.clear();
238 void setConstantInsertionPointToStart(
Block *block) {
239 constantInsertionBlock = block;
240 constantInsertionOp =
nullptr;
245 LogicalResult convertGlobal(llvm::GlobalVariable *globalVar);
247 LogicalResult convertGlobalCtorsAndDtors(llvm::GlobalVariable *globalVar);
249 FlatSymbolRefAttr getPersonalityAsAttr(llvm::Function *func);
251 LogicalResult processBasicBlock(llvm::BasicBlock *bb, Block *block);
254 LogicalResult processDebugIntrinsics();
256 LogicalResult processDebugIntrinsic(llvm::DbgVariableIntrinsic *dbgIntr,
257 DominanceInfo &domInfo);
260 LogicalResult convertIntrinsic(llvm::CallInst *inst);
263 LogicalResult convertInstruction(llvm::Instruction *inst);
268 void setNonDebugMetadataAttrs(llvm::Instruction *inst, Operation *op);
272 LogicalResult processInstruction(llvm::Instruction *inst);
277 LogicalResult convertBranchArgs(llvm::Instruction *branch,
278 llvm::BasicBlock *target,
279 SmallVectorImpl<Value> &blockArguments);
283 LogicalResult convertCallTypeAndOperands(llvm::CallBase *callInst,
284 SmallVectorImpl<Type> &types,
285 SmallVectorImpl<Value> &operands);
288 void convertParameterAttributes(llvm::Function *func, LLVMFuncOp funcOp,
292 DictionaryAttr convertParameterAttribute(llvm::AttributeSet llvmParamAttrs,
297 Type getBuiltinTypeForAttr(Type type);
301 Attribute getConstantAsAttr(llvm::Constant *constant);
304 SetVector<llvm::Constant *> getConstantsToConvert(llvm::Constant *constant);
309 FailureOr<Value> convertConstant(llvm::Constant *constant);
315 FailureOr<Value> convertConstantExpr(llvm::Constant *constant);
319 ComdatOp getGlobalComdatOp();
325 LogicalResult processTBAAMetadata(
const llvm::MDNode *node);
330 LogicalResult processAccessGroupMetadata(
const llvm::MDNode *node);
335 LogicalResult processAliasScopeMetadata(
const llvm::MDNode *node);
339 void processComdat(
const llvm::Comdat *comdat);
344 Block *constantInsertionBlock =
nullptr;
346 Operation *constantInsertionOp =
nullptr;
348 Operation *globalInsertionOp =
nullptr;
350 ComdatOp globalComdatOp =
nullptr;
352 MLIRContext *context;
356 std::unique_ptr<llvm::Module> llvmModule;
360 LLVMImportInterface iface;
363 DenseMap<llvm::BasicBlock *, Block *> blockMapping;
365 DenseMap<llvm::Value *, Value> valueMapping;
369 DenseMap<llvm::Instruction *, Operation *> noResultOpMapping;
372 SetVector<llvm::Instruction *> debugIntrinsics;
375 DenseMap<const llvm::MDNode *, Attribute> aliasScopeMapping;
378 DenseMap<const llvm::MDNode *, Attribute> tbaaMapping;
381 DenseMap<const llvm::Comdat *, SymbolRefAttr> comdatMapping;
383 LLVM::TypeFromLLVMIRTranslator typeTranslator;
385 std::unique_ptr<detail::DebugImporter> debugImporter;
387 std::unique_ptr<detail::LoopAnnotationImporter> loopAnnotationImporter;
392 bool emitExpensiveWarnings;
Attributes are known-constant values of operations.
Block represents an ordered list of Operations.
This class provides support for representing a failure result, or a valid value of type T.
LogicalResult initializeImport()
Queries all registered dialect interfaces for the supported LLVM IR intrinsic and metadata kinds and ...
Module import implementation class that provides methods to import globals and functions from an LLVM...
Location translateLoc(llvm::DILocation *loc)
Translates the debug location.
LogicalResult convertComdats()
Converts all comdat selectors of the LLVM module to MLIR comdat operations.
LogicalResult convertFunctions()
Converts all functions of the LLVM module to MLIR functions.
FailureOr< SmallVector< Value > > convertValues(ArrayRef< llvm::Value * > values)
Converts a range of LLVM values to a range of MLIR values using the convertValue method,...
Attribute lookupTBAAAttr(const llvm::MDNode *node) const
Returns the MLIR attribute mapped to the given LLVM TBAA metadata node.
ModuleImport(ModuleOp mlirModule, std::unique_ptr< llvm::Module > llvmModule, bool emitExpensiveWarnings)
DILocalVariableAttr matchLocalVariableAttr(llvm::Value *value)
Converts value to a local variable attribute.
LogicalResult convertLinkerOptionsMetadata()
Converts !llvm.linker.options metadata to the llvm.linker.options LLVM dialect operation.
void mapBlock(llvm::BasicBlock *llvm, Block *mlir)
Stores the mapping between an LLVM block and its MLIR counterpart.
void processFunctionAttributes(llvm::Function *func, LLVMFuncOp funcOp)
Converts function attributes of LLVM Function func into LLVM dialect attributes of LLVMFuncOp funcOp.
LogicalResult convertMetadata()
Converts all LLVM metadata nodes that translate to attributes such as alias analysis or access group ...
LogicalResult convertIntrinsicArguments(ArrayRef< llvm::Value * > values, 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.
Value & mapValue(llvm::Value *value)
Provides write-once access to store the MLIR value corresponding to the given LLVM value.
FailureOr< Value > convertValue(llvm::Value *value)
Converts an LLVM value to an MLIR value, or returns failure if the conversion fails.
LogicalResult initializeImportInterface()
Calls the LLVMImportInterface initialization that queries the registered dialect interfaces for the s...
void addDebugIntrinsic(llvm::CallInst *intrinsic)
Adds a debug intrinsics to the list of intrinsics that should be converted after the function convers...
Block * lookupBlock(llvm::BasicBlock *block) const
Returns the MLIR block mapped to the given LLVM block.
FailureOr< Value > convertMetadataValue(llvm::Value *value)
Converts an LLVM metadata value to an MLIR value, or returns failure if the conversion fails.
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.
void mapNoResultOp(llvm::Instruction *llvm, Operation *mlir)
Stores a mapping between an LLVM instruction and the imported MLIR operation if the operation returns...
Type convertType(llvm::Type *type)
Converts the type from LLVM to MLIR LLVM dialect.
Operation * lookupOperation(llvm::Instruction *inst)
Returns the MLIR operation mapped to the given LLVM instruction.
DILabelAttr matchLabelAttr(llvm::Value *value)
Converts value to a label attribute. Asserts if the matching fails.
FloatAttr matchFloatAttr(llvm::Value *value)
Converts value to a float attribute. Asserts if the matching fails.
LoopAnnotationAttr translateLoopAnnotationAttr(const llvm::MDNode *node, Location loc) const
Returns the loop annotation attribute that corresponds to the given LLVM loop metadata node.
void setFastmathFlagsAttr(llvm::Instruction *inst, Operation *op) const
Sets the fastmath flags attribute for the imported operation op given the original instruction inst.
FailureOr< SmallVector< AliasScopeAttr > > matchAliasScopeAttrs(llvm::Value *value)
Converts value to an array of alias scopes or returns failure if the conversion fails.
Value lookupValue(llvm::Value *value)
Returns the MLIR value mapped to the given LLVM value.
LogicalResult processFunction(llvm::Function *func)
Imports func into the current module.
Operation *& mapNoResultOp(llvm::Instruction *inst)
Provides write-once access to store the MLIR operation corresponding to the given LLVM instruction if...
void mapValue(llvm::Value *llvm, Value mlir)
Stores the mapping between an LLVM value and its MLIR counterpart.
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...
LogicalResult convertGlobals()
Converts all global variables of the LLVM module to MLIR global variables.
LogicalResult convertDataLayout()
Converts the data layout of the LLVM module to an MLIR data layout specification.
IntegerAttr matchIntegerAttr(llvm::Value *value)
Converts value to an integer attribute. Asserts if the matching fails.
Type translateType(llvm::Type *type)
Translates the given LLVM IR type to the MLIR LLVM dialect.
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.
Instances of the Type class are uniqued, have an immutable identifier and an optional mutable compone...
This class represents an instance of an SSA value in the MLIR system, representing a computable value...
Include the generated interface declarations.
Include the generated interface declarations.
This class represents an efficient way to signal success or failure.