1 | //===----- ExampleModules.h - IR modules for LLJIT examples -----*- C++ -*-===// |
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 | // Example modules for LLJIT examples |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | |
13 | #ifndef LLVM_EXAMPLES_ORCV2EXAMPLES_EXAMPLEMODULES_H |
14 | #define LLVM_EXAMPLES_ORCV2EXAMPLES_EXAMPLEMODULES_H |
15 | |
16 | #include "llvm/ADT/StringRef.h" |
17 | #include "llvm/ExecutionEngine/Orc/ThreadSafeModule.h" |
18 | #include "llvm/IR/LLVMContext.h" |
19 | #include "llvm/IR/Module.h" |
20 | #include "llvm/IRReader/IRReader.h" |
21 | #include "llvm/Support/CommandLine.h" |
22 | #include "llvm/Support/Error.h" |
23 | #include "llvm/Support/SourceMgr.h" |
24 | |
25 | const llvm::StringRef Add1Example = |
26 | R"( |
27 | define i32 @add1(i32 %x) { |
28 | entry: |
29 | %r = add nsw i32 %x, 1 |
30 | ret i32 %r |
31 | } |
32 | )" ; |
33 | |
34 | inline llvm::Error createSMDiagnosticError(llvm::SMDiagnostic &Diag) { |
35 | using namespace llvm; |
36 | std::string Msg; |
37 | { |
38 | raw_string_ostream OS(Msg); |
39 | Diag.print(ProgName: "" , S&: OS); |
40 | } |
41 | return make_error<StringError>(Args: std::move(Msg), Args: inconvertibleErrorCode()); |
42 | } |
43 | |
44 | inline llvm::Expected<llvm::orc::ThreadSafeModule> |
45 | parseExampleModule(llvm::StringRef Source, llvm::StringRef Name) { |
46 | using namespace llvm; |
47 | auto Ctx = std::make_unique<LLVMContext>(); |
48 | SMDiagnostic Err; |
49 | if (auto M = parseIR(Buffer: MemoryBufferRef(Source, Name), Err, Context&: *Ctx)) |
50 | return orc::ThreadSafeModule(std::move(M), std::move(Ctx)); |
51 | |
52 | return createSMDiagnosticError(Diag&: Err); |
53 | } |
54 | |
55 | inline llvm::Expected<llvm::orc::ThreadSafeModule> |
56 | parseExampleModuleFromFile(llvm::StringRef FileName) { |
57 | using namespace llvm; |
58 | auto Ctx = std::make_unique<LLVMContext>(); |
59 | SMDiagnostic Err; |
60 | |
61 | if (auto M = parseIRFile(Filename: FileName, Err, Context&: *Ctx)) |
62 | return orc::ThreadSafeModule(std::move(M), std::move(Ctx)); |
63 | |
64 | return createSMDiagnosticError(Diag&: Err); |
65 | } |
66 | |
67 | #endif // LLVM_EXAMPLES_ORCV2EXAMPLES_EXAMPLEMODULES_H |
68 | |