Browse Source

!13540 [MD] release gil in pynative && GPU

From: @liyong126
Reviewed-by: @liucunwei,@heleiwang
Signed-off-by: @liucunwei
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
b5396bc1cd
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/ccsrc/backend/session/executor.cc

+ 3
- 1
mindspore/ccsrc/backend/session/executor.cc View File

@@ -415,8 +415,10 @@ void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const Gr
// Release GIL before calling into (potentially long-running) C++ code
if (Py_IsInitialized()) {
py::gil_scoped_release release;
session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask);
} else {
session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask);
}
session->RunOpImpl(graph_info, op_run_info, input_tensors, outputs, tensors_mask);
}
} else {
auto task = std::make_shared<RunOpTask>();


Loading…
Cancel
Save