13 #ifndef MLIR_TARGET_LLVMIR_LLVMIMPORTINTERFACE_H
14 #define MLIR_TARGET_LLVMIR_LLVMIMPORTINTERFACE_H
22 #include "llvm/IR/Instruction.h"
23 #include "llvm/IR/Instructions.h"
24 #include "llvm/Support/FormatVariadic.h"
107 const auto *intrinsicIt =
109 return intrinsicToDialect.count(id);
115 "expected unique conversion for intrinsic ({0}), but "
116 "got conflicting {1} and {2} conversions",
117 *intrinsicIt, iface.getDialect()->getNamespace(),
118 intrinsicToDialect.lookup(*intrinsicIt)->getNamespace()));
120 const auto *instructionIt =
122 return instructionToDialect.count(id);
128 "expected unique conversion for instruction ({0}), but "
129 "got conflicting {1} and {2} conversions",
130 *intrinsicIt, iface.getDialect()->getNamespace(),
131 instructionToDialect.lookup(*intrinsicIt)
137 intrinsicToDialect[id] = iface.getDialect();
140 instructionToDialect[id] = &iface;
143 metadataToDialect[kind].push_back(iface.getDialect());
154 Dialect *dialect = intrinsicToDialect.lookup(inst->getIntrinsicID());
160 assert(iface &&
"expected to find a dialect interface");
167 return intrinsicToDialect.count(
id);
177 instructionToDialect.lookup(inst->getOpcode());
187 return instructionToDialect.count(
id);
198 auto it = metadataToDialect.find(kind);
199 if (it == metadataToDialect.end())
203 bool isSuccess =
false;
204 for (
Dialect *dialect : it->getSecond()) {
206 assert(iface &&
"expected to find a dialect interface");
213 return success(isSuccess);
219 return metadataToDialect.count(kind);
A collection of dialect interfaces within a context, for a given concrete interface type.
DialectInterfaceCollection< LLVMImportDialectInterface > Base
const LLVMImportDialectInterface * getInterfaceFor(Object *obj) const
Get the interface for a given object, or null if one is not registered.
Dialects are groups of MLIR operations, types and attributes, as well as behavior associated with the...
Base class for dialect interfaces used to import LLVM IR.
virtual LogicalResult convertIntrinsic(OpBuilder &builder, llvm::CallInst *inst, LLVM::ModuleImport &moduleImport) const
Hook for derived dialect interfaces to implement the import of intrinsics into MLIR.
LLVMImportDialectInterface(Dialect *dialect)
virtual ArrayRef< unsigned > getSupportedIntrinsics() const
Hook for derived dialect interfaces to publish the supported intrinsics.
virtual ArrayRef< unsigned > getSupportedInstructions() const
Hook for derived dialect interfaces to publish the supported instructions.
virtual LogicalResult convertInstruction(OpBuilder &builder, llvm::Instruction *inst, ArrayRef< llvm::Value * > llvmOperands, LLVM::ModuleImport &moduleImport) const
Hook for derived dialect interfaces to implement the import of instructions into MLIR.
virtual ArrayRef< unsigned > getSupportedMetadata() const
Hook for derived dialect interfaces to publish the supported metadata kinds.
virtual LogicalResult setMetadataAttrs(OpBuilder &builder, unsigned kind, llvm::MDNode *node, Operation *op, LLVM::ModuleImport &moduleImport) const
Hook for derived dialect interfaces to implement the import of metadata into MLIR.
Interface collection for the import of LLVM IR that dispatches to a concrete dialect interface implem...
LogicalResult initializeImport()
Queries all registered dialect interfaces for the supported LLVM IR intrinsic and metadata kinds and ...
LogicalResult convertInstruction(OpBuilder &builder, llvm::Instruction *inst, ArrayRef< llvm::Value * > llvmOperands, LLVM::ModuleImport &moduleImport) const
Converts the LLVM instruction to an MLIR operation if a conversion exists.
LogicalResult convertIntrinsic(OpBuilder &builder, llvm::CallInst *inst, LLVM::ModuleImport &moduleImport) const
Converts the LLVM intrinsic to an MLIR operation if a conversion exists.
LogicalResult setMetadataAttrs(OpBuilder &builder, unsigned kind, llvm::MDNode *node, Operation *op, LLVM::ModuleImport &moduleImport) const
Attaches the given LLVM metadata to the imported operation if a conversion to one or more MLIR dialec...
bool isConvertibleMetadata(unsigned kind)
Returns true if the given LLVM IR metadata is convertible to an MLIR attribute.
bool isConvertibleInstruction(unsigned id)
Returns true if the given LLVM IR instruction is convertible to an MLIR operation.
bool isConvertibleIntrinsic(llvm::Intrinsic::ID id)
Returns true if the given LLVM IR intrinsic is convertible to an MLIR operation.
Module import implementation class that provides methods to import globals and functions from an LLVM...
This class helps build Operations.
Operation is the basic unit of execution within MLIR.
The base class used for all derived interface types.
Include the generated interface declarations.
Include the generated interface declarations.
InFlightDiagnostic emitError(Location loc)
Utility method to emit an error message using this location.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...