Browse Source

fix dataset train gil of gpu

tags/v0.7.0-beta
panfengfeng 5 years ago
parent
commit
48ab208148
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindspore/ccsrc/backend/session/gpu_session.cc

+ 4
- 1
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);


Loading…
Cancel
Save