#include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h"
#include "mlir/IR/BuiltinDialect.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/Target/LLVMIR/LLVMTranslationInterface.h"
using namespace mlir;
namespace {
class BuiltinDialectLLVMIRTranslationInterface
: public LLVMTranslationDialectInterface {
public:
using LLVMTranslationDialectInterface::LLVMTranslationDialectInterface;
LogicalResult
convertOperation(Operation *op, llvm::IRBuilderBase &builder,
LLVM::ModuleTranslation &moduleTranslation) const override {
return success(isa<ModuleOp>(op));
}
};
}
void mlir::registerBuiltinDialectTranslation(DialectRegistry ®istry) {
registry.addExtension(+[](MLIRContext *ctx, BuiltinDialect *dialect) {
dialect->addInterfaces<BuiltinDialectLLVMIRTranslationInterface>();
});
}
void mlir::registerBuiltinDialectTranslation(MLIRContext &context) {
DialectRegistry registry;
registerBuiltinDialectTranslation(registry);
context.appendDialectRegistry(registry);
}