From dd4c78c03d7c2d99432444b6cddd07035db0345d Mon Sep 17 00:00:00 2001
From: kswang <wangkaisheng2@huawei.com>
Date: Mon, 26 Oct 2020 16:52:47 +0800
Subject: [PATCH] notify lockedtensor before output

---
 mindspore/ccsrc/backend/session/executor.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/backend/session/executor.cc b/mindspore/ccsrc/backend/session/executor.cc
index 06dbd6c0e2..46fd50bfc1 100644
--- a/mindspore/ccsrc/backend/session/executor.cc
+++ b/mindspore/ccsrc/backend/session/executor.cc
@@ -109,11 +109,10 @@ void RunGraphTask::Run() {
   } catch (const std::exception &e) {
     MsException::GetInstance().SetException();
   }
-
-  NotifyOutputTensors(&outputs_);
   for (auto &tensor : input_need_lock_tensors_) {
     tensor->SetNeedWait(false);
   }
+  NotifyOutputTensors(&outputs_);
   ExecutorManager::Instance().OnRunGraphFinished();
 }