1 | //===- ConvertFromLLVMIR.cpp - MLIR to LLVM IR conversion -----------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // This file implements the function that registers the translation between |
10 | // LLVM IR and the MLIR LLVM dialect. |
11 | // |
12 | //===----------------------------------------------------------------------===// |
13 | |
14 | #include "mlir/Dialect/DLTI/DLTI.h" |
15 | #include "mlir/IR/BuiltinOps.h" |
16 | #include "mlir/Target/LLVMIR/Dialect/All.h" |
17 | #include "mlir/Target/LLVMIR/Import.h" |
18 | #include "mlir/Tools/mlir-translate/Translation.h" |
19 | #include "llvm/IR/Module.h" |
20 | #include "llvm/IR/Verifier.h" |
21 | #include "llvm/IRReader/IRReader.h" |
22 | #include "llvm/Support/SourceMgr.h" |
23 | |
24 | using namespace mlir; |
25 | |
26 | namespace mlir { |
27 | void registerFromLLVMIRTranslation() { |
28 | static llvm::cl::opt<bool> emitExpensiveWarnings( |
29 | "emit-expensive-warnings" , |
30 | llvm::cl::desc("Emit expensive warnings during LLVM IR import " |
31 | "(discouraged: testing only!)" ), |
32 | llvm::cl::init(Val: false)); |
33 | static llvm::cl::opt<bool> dropDICompositeTypeElements( |
34 | "drop-di-composite-type-elements" , |
35 | llvm::cl::desc( |
36 | "Avoid translating the elements of DICompositeTypes during " |
37 | "the LLVM IR import (discouraged: testing only!)" ), |
38 | llvm::cl::init(Val: false)); |
39 | |
40 | static llvm::cl::opt<bool> preferUnregisteredIntrinsics( |
41 | "prefer-unregistered-intrinsics" , |
42 | llvm::cl::desc( |
43 | "Prefer translating all intrinsics into llvm.call_intrinsic instead " |
44 | "of using dialect supported intrinsics" ), |
45 | llvm::cl::init(Val: false)); |
46 | |
47 | static llvm::cl::opt<bool> importStructsAsLiterals( |
48 | "import-structs-as-literals" , |
49 | llvm::cl::desc("Controls if structs should be imported as literal " |
50 | "structs, i.e., nameless structs." ), |
51 | llvm::cl::init(Val: false)); |
52 | |
53 | TranslateToMLIRRegistration registration( |
54 | "import-llvm" , "Translate LLVMIR to MLIR" , |
55 | [](llvm::SourceMgr &sourceMgr, |
56 | MLIRContext *context) -> OwningOpRef<Operation *> { |
57 | llvm::SMDiagnostic err; |
58 | llvm::LLVMContext llvmContext; |
59 | std::unique_ptr<llvm::Module> llvmModule = |
60 | llvm::parseIR(Buffer: *sourceMgr.getMemoryBuffer(i: sourceMgr.getMainFileID()), |
61 | Err&: err, Context&: llvmContext); |
62 | if (!llvmModule) { |
63 | std::string errStr; |
64 | llvm::raw_string_ostream errStream(errStr); |
65 | err.print(/*ProgName=*/"" , S&: errStream); |
66 | emitError(UnknownLoc::get(context)) << errStr; |
67 | return {}; |
68 | } |
69 | if (llvm::verifyModule(M: *llvmModule, OS: &llvm::errs())) |
70 | return nullptr; |
71 | |
72 | // Debug records are not currently supported in the LLVM IR translator. |
73 | if (llvmModule->IsNewDbgInfoFormat) |
74 | llvmModule->convertFromNewDbgValues(); |
75 | |
76 | return translateLLVMIRToModule( |
77 | std::move(llvmModule), context, emitExpensiveWarnings, |
78 | dropDICompositeTypeElements, /*loadAllDialects=*/true, |
79 | preferUnregisteredIntrinsics, importStructsAsLiterals); |
80 | }, |
81 | [](DialectRegistry ®istry) { |
82 | // Register the DLTI dialect used to express the data layout |
83 | // specification of the imported module. |
84 | registry.insert<DLTIDialect>(); |
85 | // Register all dialects that implement the LLVMImportDialectInterface |
86 | // including the LLVM dialect. |
87 | registerAllFromLLVMIRTranslations(registry); |
88 | }); |
89 | } |
90 | } // namespace mlir |
91 | |