fix py_transform gpu hang problem

5 years ago · 8b585c76e2
parent 9fe6074c3f
commit 8b585c76e2
1 changed files with 5 additions and 2 deletions
--- a/mindspore/ccsrc/session/gpu_session.cc
+++ b/mindspore/ccsrc/session/gpu_session.cc
@ -125,8 +125,11 @@ void GPUSession::RunGraph(const GraphId &graph_id, const std::vector<tensor::Ten
  MS_EXCEPTION_IF_NULL(kernel_graph);
  // Convert inputs to model
  predictmodel::StepConvertWeight(inputs);
+  {
+    py::gil_scoped_release gil_release;
    // Run graph on GPU
    Execute(kernel_graph);
+  }
  // Get result from GPU
  UpdateOutputs(kernel_graph, outputs, inputs);
  // Summary