[Orc] Revert r236465 - It broke the Windows bots.

[oota-llvm.git] / examples / Kaleidoscope / Orc / fully_lazy / toy.cpp

diff --git a/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp b/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp
index 93de33353daeac5ae5bf6041b908fd394ed72232..b5822921b400115ee8552c6b47b2829af1ee7aa9 100644 (file)
--- a/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp
+++ b/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp
@@ -1214,11 +1214,11 @@ public:
   void removeModule(ModuleHandleT H) { LazyEmitLayer.removeModuleSet(H); }
 
   JITSymbol findSymbol(const std::string &Name) {
-    return LazyEmitLayer.findSymbol(Name, false);
+    return LazyEmitLayer.findSymbol(Name, true);
   }
 
   JITSymbol findSymbolIn(ModuleHandleT H, const std::string &Name) {
-    return LazyEmitLayer.findSymbolIn(H, Name, false);
+    return LazyEmitLayer.findSymbolIn(H, Name, true);
   }
 
   JITSymbol findUnmangledSymbol(const std::string &Name) {
@@ -1276,7 +1276,7 @@ private:
     makeStub(*F, *FunctionBodyPointer);
 
     // Step 4) Add the module containing the stub to the JIT.
-    auto StubH = addModule(C.takeM());
+    auto H = addModule(C.takeM());
 
     // Step 5) Set the compile and update actions.
     //
@@ -1289,20 +1289,14 @@ private:
     //   The update action will update FunctionBodyPointer to point at the newly
     // compiled function.
     std::shared_ptr<FunctionAST> Fn = std::move(FnAST);
-    CallbackInfo.setCompileAction([this, Fn, BodyPtrName, StubH]() {
+    CallbackInfo.setCompileAction([this, Fn]() {
       auto H = addModule(IRGen(Session, *Fn));
-      auto BodySym = findUnmangledSymbolIn(H, Fn->Proto->Name);
-      auto BodyPtrSym = findUnmangledSymbolIn(StubH, BodyPtrName);
-      assert(BodySym && "Missing function body.");
-      assert(BodyPtrSym && "Missing function pointer.");
-      auto BodyAddr = BodySym.getAddress();
-      auto BodyPtr = reinterpret_cast<void*>(
-                       static_cast<uintptr_t>(BodyPtrSym.getAddress()));
-      memcpy(BodyPtr, &BodyAddr, sizeof(uintptr_t));
-      return BodyAddr;
+      return findUnmangledSymbolIn(H, Fn->Proto->Name).getAddress();
     });
+    CallbackInfo.setUpdateAction(
+        getLocalFPUpdater(LazyEmitLayer, H, mangle(BodyPtrName)));
 
-    return StubH;
+    return H;
   }
 
   SessionContext &Session;