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

!13083 [MD] [GPU] fix pynative get data timeout 

From: @liyong126
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-03-12 17:00:21 +08:00 committed by Gitee
parent c35d2d0899 256e0f7c07
commit 0cabffe945
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
View File

@ -1490,7 +1490,11 @@ py::object PynativeExecutor::RunOpInMs(const OpExecInfoPtr &op_exec_info, Pynati
op_exec_info->next_input_index};
#endif
VectorRef outputs;
session->RunOp(&op_run_info, graph_info, &input_tensors, &outputs, tensors_mask);
{
// Release GIL before calling into (potentially long-running) C++ code
py::gil_scoped_release release;
session->RunOp(&op_run_info, graph_info, &input_tensors, &outputs, tensors_mask);
}
if (op_exec_info->is_dynamic_shape) {
op_exec_info->abstract = op_run_info.abstract;
}