[Orc][Kaleidoscope] Fix a missed symbol mangling operation in the fully lazy tutorial.

[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 333cb94e36509597f4120fe41c198c08859b6ee6..2e65756049f23281caafe5333c79c8d9cf4d7dfd 100644 (file)
--- a/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp
+++ b/examples/Kaleidoscope/Orc/fully_lazy/toy.cpp
@@ -1223,6 +1223,10 @@ public:
     return findSymbol(mangle(Name));
   }
 
+  JITSymbol findUnmangledSymbolIn(ModuleHandleT H, const std::string &Name) {
+    return findSymbolIn(H, mangle(Name));
+  }
+
 private:
 
   // This method searches the FunctionDefs map for a definition of 'Name'. If it
@@ -1284,7 +1288,7 @@ private:
     std::shared_ptr<FunctionAST> Fn = std::move(FnAST);
     CallbackInfo.setCompileAction([this, Fn]() {
       auto H = addModule(IRGen(Session, *Fn));
-      return findSymbolIn(H, Fn->Proto->Name).getAddress();
+      return findUnmangledSymbolIn(H, Fn->Proto->Name).getAddress();
     });
     CallbackInfo.setUpdateAction(
       CompileCallbacks.getLocalFPUpdater(H, mangle(BodyPtrName)));