fix test

apache · May 15, 2018 · 4c1c1f3 · 4c1c1f3
1 parent e3f7dfe
commit 4c1c1f3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/engine/threaded_engine_pooled.cc b/src/engine/threaded_engine_pooled.cc
@@ -57,9 +57,9 @@ class ThreadedEnginePooled : public ThreadedEngine {
     task_queue_->SignalForKill();
     task_queue_ = nullptr;
     io_task_queue_->SignalForKill();
-    io_task_queue = nullptr;
-    thread_pool_.reset(nullptr);
-    io_thread_pool = nullptr;
+    io_task_queue_ = nullptr;
+    thread_pool_ = nullptr;
+    io_thread_pool_ = nullptr;
   }
 
   void Stop() override {

diff --git a/tests/cpp/engine/threaded_engine_test.cc b/tests/cpp/engine/threaded_engine_test.cc
@@ -129,8 +129,8 @@ TEST(Engine, start_stop) {
   engine[2] = mxnet::engine::CreateThreadedEnginePerDevice();
 
   for (int i = 0; i < num_engine; ++i) {
-    engine[i].stop();
-    engine[i].start();
+    engine[i]->Stop();
+    engine[i]->Start();
   }
 }