Add support for atexit function, remove support for __main function

llvm-svn: 6194

1 files changed, 2 insertions, 1 deletions

diff --git a/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp b/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
index 0ed0f04b15b..3f4493d19be 100644
--- a/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
+++ b/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
@@ -115,7 +115,7 @@ void Interpreter::handleUserInput() {
     case Call:
       std::cin >> Command;
       callFunction(Command);    // Enter the specified function
-      finish();               // Run until it's complete
+      finish();                 // Run until it's complete
       break;
 
     case TraceOpt:
@@ -129,6 +129,7 @@ void Interpreter::handleUserInput() {
     }
 
   } while (!UserQuit);
+  AtExitHandlers.clear();
 }
 
 //===----------------------------------------------------------------------===//