MLIR  19.0.0git
ConvertFromLLVMIR.cpp
Go to the documentation of this file.
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"
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 {
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(false));
33 
34  TranslateToMLIRRegistration registration(
35  "import-llvm", "Translate LLVMIR to MLIR",
36  [](llvm::SourceMgr &sourceMgr,
38  llvm::SMDiagnostic err;
39  llvm::LLVMContext llvmContext;
40  std::unique_ptr<llvm::Module> llvmModule =
41  llvm::parseIR(*sourceMgr.getMemoryBuffer(sourceMgr.getMainFileID()),
42  err, llvmContext);
43  if (!llvmModule) {
44  std::string errStr;
45  llvm::raw_string_ostream errStream(errStr);
46  err.print(/*ProgName=*/"", errStream);
47  emitError(UnknownLoc::get(context)) << errStream.str();
48  return {};
49  }
50  if (llvm::verifyModule(*llvmModule, &llvm::errs()))
51  return nullptr;
52 
53  return translateLLVMIRToModule(std::move(llvmModule), context,
54  emitExpensiveWarnings);
55  },
56  [](DialectRegistry &registry) {
57  // Register the DLTI dialect used to express the data layout
58  // specification of the imported module.
59  registry.insert<DLTIDialect>();
60  // Register all dialects that implement the LLVMImportDialectInterface
61  // including the LLVM dialect.
63  });
64 }
65 } // namespace mlir
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
MLIRContext is the top-level object for a collection of MLIR operations.
Definition: MLIRContext.h:60
This class acts as an owning reference to an op, and will automatically destroy the held op on destru...
Definition: OwningOpRef.h:28
Include the generated interface declarations.
OwningOpRef< ModuleOp > translateLLVMIRToModule(std::unique_ptr< llvm::Module > llvmModule, MLIRContext *context, bool emitExpensiveWarnings=true)
Translates the LLVM module into an MLIR module living in the given context.
InFlightDiagnostic emitError(Location loc)
Utility method to emit an error message using this location.
static void registerAllFromLLVMIRTranslations(DialectRegistry &registry)
Registers all dialects that can be translated from LLVM IR and the corresponding translation interfac...
Definition: All.h:78
void registerFromLLVMIRTranslation()
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...
Use Translate[ToMLIR|FromMLIR]Registration as an initializer that registers a function and associates...
Definition: Translation.h:110