Fixed a memory leak issue in InternalThread (and removed caffe::Thread calss).

1 files changed, 1 insertions, 7 deletions

diff --git a/src/caffe/internal_thread.cpp b/src/caffe/internal_thread.cpp
index d7b6ae20..62a4c1fa 100644
--- a/src/caffe/internal_thread.cpp
+++ b/src/caffe/internal_thread.cpp
@@ -1,14 +1,9 @@
 #include "caffe/internal_thread.hpp"
 
-#include "caffe/util/thread.hpp"
-
 namespace caffe {
 
 InternalThread::~InternalThread() {
   WaitForInternalThreadToExit();
-  if (thread_ != NULL) {
-    delete thread_;
-  }
 }
 
 bool InternalThread::StartInternalThread() {
@@ -16,8 +11,7 @@ bool InternalThread::StartInternalThread() {
     return false;
   }
   try {
-    thread_ = new caffe::Thread
-        (&InternalThread::InternalThreadEntry, this);
+    thread_.reset(new boost::thread(&InternalThread::InternalThreadEntry, this));
   } catch (...) {
     return false;
   }