p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix npu client release problem

This commit is contained in:
yeyunpeng 2021-01-04 16:47:41 +08:00
parent 9d03a7d802
commit 98eb115731
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/src/runtime/agent/npu/npu_executor.cc
View File

@ -20,6 +20,7 @@
#include "nnacl/pack.h" #include "nnacl/pack.h"
namespace mindspore::lite { namespace mindspore::lite {
NPUExecutor::~NPUExecutor() { NPUExecutor::~NPUExecutor() {
client_.reset();
npu_input_tensors_.clear(); npu_input_tensors_.clear();
npu_output_tensors_.clear(); npu_output_tensors_.clear();
} }

1
mindspore/lite/src/runtime/agent/npu/npu_manager.cc
View File

@ -67,6 +67,7 @@ void NPUManager::Reset() {
model->is_freed_ = true; model->is_freed_ = true;
model->desc_.reset(); model->desc_.reset();
model->desc_ = nullptr; model->desc_ = nullptr;
model->client_.reset();
} }
} }
models_.clear(); models_.clear();