notify lockedtensor before output

2020-10-26 16:52:47 +08:00 · 2020-10-26 16:52:47 +08:00 · dd4c78c03d
parent 4bbb854d3c
commit dd4c78c03d
1 changed files with 1 additions and 2 deletions
--- 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();
 }