diff --git a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc index 9466f471ad..1655ab5e31 100644 --- a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc +++ b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc @@ -15,6 +15,7 @@ */ #include "runtime/device/gpu/gpu_kernel_runtime.h" #include +#include "pybind11/pybind11.h" #include "runtime/device/gpu/gpu_device_address.h" #include "runtime/device/gpu/cuda_driver.h" #include "runtime/device/gpu/gpu_buffer_mgr.h" @@ -40,6 +41,7 @@ namespace mindspore { namespace device { namespace gpu { +namespace py = pybind11; using mindspore::device::memswap::MemSwapInfoSet; using mindspore::device::memswap::MemSwapManager; using mindspore::device::memswap::SwapKind; @@ -468,6 +470,7 @@ bool GPUKernelRuntime::Run(session::KernelGraph *graph, bool is_task_sink, Debug ret = RunOneStep(graph, debugger); } else { + py::gil_scoped_release gil_release; ret = LaunchKernel(graph); } (void)gettimeofday(&end_time, nullptr);