p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

notify lockedtensor before output

This commit is contained in:
kswang 2020-10-26 16:52:47 +08:00
parent 4bbb854d3c
commit dd4c78c03d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/session/executor.cc
View File

@ -109,11 +109,10 @@ void RunGraphTask::Run() {
} catch (const std::exception &e) { } catch (const std::exception &e) {
MsException::GetInstance().SetException(); MsException::GetInstance().SetException();
} }
NotifyOutputTensors(&outputs_);
for (auto &tensor : input_need_lock_tensors_) { for (auto &tensor : input_need_lock_tensors_) {
tensor->SetNeedWait(false); tensor->SetNeedWait(false);
} }
NotifyOutputTensors(&outputs_);
ExecutorManager::Instance().OnRunGraphFinished(); ExecutorManager::Instance().OnRunGraphFinished();
} }