Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 7df16cb

Browse files
waytrue17Wei Chu
andauthored
remove cleanup on side threads (#19557)
Co-authored-by: Wei Chu <weichu@amazon.com>
1 parent 2e86b92 commit 7df16cb

2 files changed

Lines changed: 0 additions & 7 deletions

File tree

src/engine/naive_engine.cc

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,14 +74,11 @@ class NaiveEngine final : public Engine {
7474
LOG(INFO) << "Engine shutdown";
7575
for (size_t i = 0; i < streams_.size(); ++i) {
7676
if (streams_[i] != nullptr) {
77-
// Catch exception for CUDA driver shutdown
78-
MSHADOW_CATCH_ERROR(mshadow::DeleteStream(streams_[i]));
7977
streams_[i] = nullptr;
8078
}
8179
}
8280
for (size_t i = 0; i < aux_streams_.size(); ++i) {
8381
if (aux_streams_[i] != nullptr) {
84-
delete aux_streams_[i];
8582
aux_streams_[i] = nullptr;
8683
}
8784
}

src/engine/threaded_engine_perdevice.cc

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -271,10 +271,6 @@ class ThreadedEnginePerDevice : public ThreadedEngine {
271271
while (task_queue->Pop(&opr_block)) {
272272
this->ExecuteOprBlock(run_ctx, opr_block);
273273
}
274-
// Catch exception for CUDA driver shutdown
275-
MSHADOW_CATCH_ERROR(mshadow::DeleteStream<gpu>(stream));
276-
if (aux_stream != nullptr)
277-
delete aux_stream;
278274
#else
279275
ready_event->signal();
280276
#endif

0 commit comments

Comments
 (0)