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

!3635 fix dataset & train gil lock of gpu process master 

Merge pull request !3635 from panfengfeng/fix_dataset_train_gil_of_gpu_master
This commit is contained in:
mindspore-ci-bot 2020-07-29 14:41:32 +08:00 committed by Gitee
parent 44e739ae31 48ab208148
commit c57ad1528f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/backend/session/gpu_session.cc
View File

@ -296,7 +296,10 @@ py::tuple GPUSession::RunOp(const OpRunInfo &op_run_info, const GraphInfo &graph
RunOpAllocateMemory(op_run_info.value, input_tensors, kernel_graph.get());
// Execute the computation
LoadInputData(kernel_graph, input_tensors);
Execute(kernel_graph);
{
py::gil_scoped_release gil_release;
Execute(kernel_graph);
}
// Fetch outputs
VectorRef outputs;
UpdateOutputs(kernel_graph, &outputs, input_tensors);