diff --git a/mindspore/ccsrc/backend/session/executor.cc b/mindspore/ccsrc/backend/session/executor.cc index 06dbd6c0e25..46fd50bfc1e 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(); }