From 9ce983b320219289cc0b4d7d17a1a53104e335d1 Mon Sep 17 00:00:00 2001 From: yefeng Date: Sat, 2 Apr 2022 17:40:42 +0800 Subject: [PATCH] fix delete bug --- mindspore/lite/src/cxx_api/model_pool/model_pool.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/cxx_api/model_pool/model_pool.cc b/mindspore/lite/src/cxx_api/model_pool/model_pool.cc index d86a6002ebd..b475ea14feb 100644 --- a/mindspore/lite/src/cxx_api/model_pool/model_pool.cc +++ b/mindspore/lite/src/cxx_api/model_pool/model_pool.cc @@ -624,7 +624,9 @@ Status ModelPool::Predict(const std::vector &inputs, std::vectorSetPredictTaskDone(); + if (predict_task_queue_ != nullptr) { + predict_task_queue_->SetPredictTaskDone(); + } for (auto &th : model_worker_vec_) { if (th.joinable()) { th.join();