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("", OS);
40   }
41   return make_error<StringError>(std::move(Msg), 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(MemoryBufferRef(Source, Name), Err, *Ctx))
50     return orc::ThreadSafeModule(std::move(M), std::move(Ctx));
51 
52   return createSMDiagnosticError(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, Err, *Ctx))
62     return orc::ThreadSafeModule(std::move(M), std::move(Ctx));
63 
64   return createSMDiagnosticError(Err);
65 }
66 
67 #endif // LLVM_EXAMPLES_ORCV2EXAMPLES_EXAMPLEMODULES_H
68