From 7df16cba2fbb58dbb9f104168c82cb7379f14e38 Mon Sep 17 00:00:00 2001 From: waytrue17 <52505574+waytrue17@users.noreply.github.com> Date: Wed, 18 Nov 2020 16:08:10 -0800 Subject: [PATCH] remove cleanup on side threads (#19557) Co-authored-by: Wei Chu --- src/engine/naive_engine.cc | 3 --- src/engine/threaded_engine_perdevice.cc | 4 ---- 2 files changed, 7 deletions(-) diff --git a/src/engine/naive_engine.cc b/src/engine/naive_engine.cc index d98f77d03bc8..8a1c22e936fa 100644 --- a/src/engine/naive_engine.cc +++ b/src/engine/naive_engine.cc @@ -74,14 +74,11 @@ class NaiveEngine final : public Engine { LOG(INFO) << "Engine shutdown"; for (size_t i = 0; i < streams_.size(); ++i) { if (streams_[i] != nullptr) { - // Catch exception for CUDA driver shutdown - MSHADOW_CATCH_ERROR(mshadow::DeleteStream(streams_[i])); streams_[i] = nullptr; } } for (size_t i = 0; i < aux_streams_.size(); ++i) { if (aux_streams_[i] != nullptr) { - delete aux_streams_[i]; aux_streams_[i] = nullptr; } } diff --git a/src/engine/threaded_engine_perdevice.cc b/src/engine/threaded_engine_perdevice.cc index 2184d784a414..440d2904a497 100644 --- a/src/engine/threaded_engine_perdevice.cc +++ b/src/engine/threaded_engine_perdevice.cc @@ -271,10 +271,6 @@ class ThreadedEnginePerDevice : public ThreadedEngine { while (task_queue->Pop(&opr_block)) { this->ExecuteOprBlock(run_ctx, opr_block); } - // Catch exception for CUDA driver shutdown - MSHADOW_CATCH_ERROR(mshadow::DeleteStream(stream)); - if (aux_stream != nullptr) - delete aux_stream; #else ready_event->signal(); #endif