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

View File

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

View File

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