Browse Source

notify lockedtensor before output

tags/v1.1.0
kswang 5 years ago
parent
commit
dd4c78c03d
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