Browse Source

!7849 run op check input tensor

Merge pull request !7849 from kisnwang/pynative-wait-input-tensor
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
59162dc2b1
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/ccsrc/backend/session/executor.cc

+ 6
- 0
mindspore/ccsrc/backend/session/executor.cc View File

@@ -220,6 +220,7 @@ bool Executor::IsTaskReady(const std::shared_ptr<RunGraphTask> &task) {
void Executor::SyncRunTask(const std::shared_ptr<Task> &task) {
std::unique_lock<std::mutex> lock(task_mutex_);
ready_tasks_.push(task);
done_tasks_.clear();
task_cond_var_.notify_all();
sync_cond_var_.wait(lock);
MsException::GetInstance().CheckException();
@@ -328,6 +329,11 @@ void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const Gr
task->op_run_info_ = op_run_info;
task->graph_info_ = graph_info;
task->input_tensors_ = input_tensors;
for (auto &tensor : input_tensors) {
if (tensor->NeedWait()) {
tensor->Wait();
}
}
SyncRunTask(task);
*outputs = task->outputs_;
}


Loading…
Cancel
Save