Browse Source

!12648 [Pynative] Remove asynchronous execution

From: @chenyijie6
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
d2d6c3cfb5
1 changed files with 2 additions and 8 deletions
  1. +2
    -8
      mindspore/ccsrc/backend/session/executor.cc

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

@@ -398,19 +398,13 @@ void Executor::RunGraphAsync(const SessionPtr &session, const GraphId &graph_id,
void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const GraphInfo &graph_info,
std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs,
const std::vector<int64_t> &tensors_mask) {
auto task = std::make_shared<RunOpTask>();
task->session_ = session;
task->op_run_info_ = op_run_info;
task->graph_info_ = graph_info;
task->input_tensors_ = input_tensors;
task->tensors_mask_ = tensors_mask;
MS_EXCEPTION_IF_NULL(session);
for (auto &tensor : *input_tensors) {
if (tensor->NeedWait()) {
tensor->Wait();
}
}
RunTask(task, true, true);
*outputs = task->outputs_;
session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask);
}

void Executor::RunOpsInGraph(const SessionPtr &session, const GraphId &graph_id,


Loading…
Cancel
Save