提交 89a86611 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1391 Fix py_transforms gpu end-to-end testcases hang problem

Merge pull request !1391 from xiefangqi/xfq_fix_gpu_issue
......@@ -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);
// Run graph on GPU
Execute(kernel_graph);
{
py::gil_scoped_release gil_release;
// Run graph on GPU
Execute(kernel_graph);
}
// Get result from GPU
UpdateOutputs(kernel_graph, outputs, inputs);
// Summary
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册