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

Delete Asynchronous Execution

This commit is contained in:
chenyijie6 2021-02-25 19:25:41 +08:00
parent b16fcc8037
commit 644770417c
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
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,