mirror of
https://github.com/intel/llvm.git
synced 2026-01-27 06:06:34 +08:00
66900b3eae
Migrate the translation of the OpenMP dialect operations to LLVM IR to the new dialect-based mechanism. Depends On D96503 Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D96504
80 lines
2.9 KiB
C++
80 lines
2.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/OpenMP/OpenMPDialect.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.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;
|
|
|
|
std::unique_ptr<llvm::Module>
|
|
mlir::translateModuleToLLVMIR(ModuleOp m, llvm::LLVMContext &llvmContext,
|
|
StringRef name) {
|
|
auto llvmModule =
|
|
LLVM::ModuleTranslation::translateModule<>(m, llvmContext, name);
|
|
if (!llvmModule)
|
|
emitError(m.getLoc(), "Fail to convert MLIR to LLVM IR");
|
|
else if (verifyModule(*llvmModule))
|
|
emitError(m.getLoc(), "LLVM IR fails to verify");
|
|
return llvmModule;
|
|
}
|
|
|
|
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 = LLVM::ModuleTranslation::translateModule<>(
|
|
module, llvmContext, "LLVMDialectModule");
|
|
if (!llvmModule)
|
|
return failure();
|
|
|
|
llvmModule->print(output, nullptr);
|
|
return success();
|
|
},
|
|
[](DialectRegistry ®istry) {
|
|
registry.insert<omp::OpenMPDialect>();
|
|
registry.addDialectInterface<omp::OpenMPDialect,
|
|
OpenMPDialectLLVMIRTranslationInterface>();
|
|
registerLLVMDialectTranslation(registry);
|
|
});
|
|
}
|
|
} // namespace mlir
|