Browse Source

!7772 notify lockedtensor before output

Merge pull request !7772 from kisnwang/fix-asynrun-segmentfault
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
a85bcdf6e3
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/ccsrc/backend/session/executor.cc

+ 1
- 2
mindspore/ccsrc/backend/session/executor.cc View File

@@ -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();
}



Loading…
Cancel
Save