From 62b20a278f32e176e98b78622fe8ec56137afa33 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Thu, 1 Jul 2021 14:57:17 +0800 Subject: [PATCH] fix npu free not set null --- mindspore/lite/src/delegate/npu/npu_delegate.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/lite/src/delegate/npu/npu_delegate.cc b/mindspore/lite/src/delegate/npu/npu_delegate.cc index a3da74eea87..4b630dfceec 100644 --- a/mindspore/lite/src/delegate/npu/npu_delegate.cc +++ b/mindspore/lite/src/delegate/npu/npu_delegate.cc @@ -77,11 +77,13 @@ int NPUDelegate::Init() { if (!npu_manager_->IsSupportNPU()) { MS_LOG(DEBUG) << "Checking that npu is unsupported."; free(npu_manager_); + npu_manager_ = nullptr; return RET_NOT_SUPPORT; } pass_manager_ = new (std::nothrow) NPUPassManager(); if (pass_manager_ == nullptr) { free(npu_manager_); + npu_manager_ = nullptr; MS_LOG(ERROR) << "New npu pass manager failed."; return RET_ERROR; }