diff options
Diffstat (limited to 'llvm')
| -rw-r--r-- | llvm/examples/Kaleidoscope/Orc/initial/toy.cpp | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/llvm/examples/Kaleidoscope/Orc/initial/toy.cpp b/llvm/examples/Kaleidoscope/Orc/initial/toy.cpp index 6daa8d5e924..1b772259a19 100644 --- a/llvm/examples/Kaleidoscope/Orc/initial/toy.cpp +++ b/llvm/examples/Kaleidoscope/Orc/initial/toy.cpp @@ -1180,16 +1180,24 @@ private: CompileLayerT CompileLayer; }; -static void HandleDefinition(SessionContext &S, KaleidoscopeJIT &J) { - if (auto F = ParseDefinition()) { - IRGenContext C(S); - if (auto LF = F->IRGen(C)) { +static std::unique_ptr<llvm::Module> +IRGen(KaleidoscopeJIT &J, SessionContext &S, const FunctionAST &F) { + IRGenContext C(S); + auto LF = F.IRGen(C); + if (!LF) + return nullptr; #ifndef MINIMAL_STDERR_OUTPUT - std::cerr << "Read function definition:\n"; - LF->dump(); + fprintf(stderr, "Read function definition:"); + LF->dump(); #endif - J.addModule(C.takeM()); + return C.takeM(); +} + +static void HandleDefinition(SessionContext &S, KaleidoscopeJIT &J) { + if (auto F = ParseDefinition()) { + if (auto M = IRGen(J, S, *F)) { S.addPrototypeAST(llvm::make_unique<PrototypeAST>(*F->Proto)); + J.addModule(std::move(M)); } } else { // Skip token for error recovery. |
