Browse Source

!10859 fix npu pad and softmax bug

From: @yeyunpeng2020
Reviewed-by: @hangangqiang,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
1f89f854aa
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/runtime/agent/npu/npu_executor.cc
  2. +1
    -0
      mindspore/lite/src/runtime/agent/npu/npu_manager.cc

+ 1
- 0
mindspore/lite/src/runtime/agent/npu/npu_executor.cc 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();
}


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

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


Loading…
Cancel
Save