ce8f10d6cb
A series of preceding patches changed the mechanism for translating MLIR to LLVM IR to use dialect interface with delayed registration. It is no longer necessary for specific dialects to derive from ModuleTranslation. Remove all virtual methods from ModuleTranslation and factor out the entry point to be a free function. Also perform some cleanups in ModuleTranslation internals. Depends On D96774 Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D96775
92 lines
3.9 KiB
C++
92 lines
3.9 KiB
C++
//===- ConvertToLLVMIR.cpp - MLIR to LLVM IR conversion -------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements a translation between the MLIR LLVM dialect and LLVM IR.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Target/LLVMIR.h"
|
|
|
|
#include "mlir/Dialect/LLVMIR/LLVMAVX512Dialect.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMArmNeonDialect.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMArmSVEDialect.h"
|
|
#include "mlir/Dialect/LLVMIR/NVVMDialect.h"
|
|
#include "mlir/Dialect/LLVMIR/ROCDLDialect.h"
|
|
#include "mlir/Dialect/OpenMP/OpenMPDialect.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMAVX512/LLVMAVX512ToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMArmNeon/LLVMArmNeonToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMArmSVE/LLVMArmSVEToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/NVVM/NVVMToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/ROCDL/ROCDLToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/ModuleTranslation.h"
|
|
#include "mlir/Translation.h"
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/Verifier.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
using namespace mlir;
|
|
|
|
void mlir::registerLLVMDialectTranslation(DialectRegistry ®istry) {
|
|
registry.insert<LLVM::LLVMDialect>();
|
|
registry.addDialectInterface<LLVM::LLVMDialect,
|
|
LLVMDialectLLVMIRTranslationInterface>();
|
|
}
|
|
|
|
void mlir::registerLLVMDialectTranslation(MLIRContext &context) {
|
|
auto *dialect = context.getLoadedDialect<LLVM::LLVMDialect>();
|
|
if (!dialect || dialect->getRegisteredInterface<
|
|
LLVMDialectLLVMIRTranslationInterface>() == nullptr) {
|
|
DialectRegistry registry;
|
|
registry.insert<LLVM::LLVMDialect>();
|
|
registry.addDialectInterface<LLVM::LLVMDialect,
|
|
LLVMDialectLLVMIRTranslationInterface>();
|
|
context.appendDialectRegistry(registry);
|
|
}
|
|
}
|
|
|
|
namespace mlir {
|
|
void registerToLLVMIRTranslation() {
|
|
TranslateFromMLIRRegistration registration(
|
|
"mlir-to-llvmir",
|
|
[](ModuleOp module, raw_ostream &output) {
|
|
llvm::LLVMContext llvmContext;
|
|
auto llvmModule = translateModuleToLLVMIR(module, llvmContext);
|
|
if (!llvmModule)
|
|
return failure();
|
|
|
|
llvmModule->print(output, nullptr);
|
|
return success();
|
|
},
|
|
[](DialectRegistry ®istry) {
|
|
registry.insert<omp::OpenMPDialect, LLVM::LLVMAVX512Dialect,
|
|
LLVM::LLVMArmSVEDialect, LLVM::LLVMArmNeonDialect,
|
|
NVVM::NVVMDialect, ROCDL::ROCDLDialect>();
|
|
registry.addDialectInterface<omp::OpenMPDialect,
|
|
OpenMPDialectLLVMIRTranslationInterface>();
|
|
registry
|
|
.addDialectInterface<LLVM::LLVMAVX512Dialect,
|
|
LLVMAVX512DialectLLVMIRTranslationInterface>();
|
|
registry.addDialectInterface<
|
|
LLVM::LLVMArmNeonDialect,
|
|
LLVMArmNeonDialectLLVMIRTranslationInterface>();
|
|
registry
|
|
.addDialectInterface<LLVM::LLVMArmSVEDialect,
|
|
LLVMArmSVEDialectLLVMIRTranslationInterface>();
|
|
registry.addDialectInterface<NVVM::NVVMDialect,
|
|
NVVMDialectLLVMIRTranslationInterface>();
|
|
registry.addDialectInterface<ROCDL::ROCDLDialect,
|
|
ROCDLDialectLLVMIRTranslationInterface>();
|
|
registerLLVMDialectTranslation(registry);
|
|
});
|
|
}
|
|
} // namespace mlir
|