!6131 fix gpu pynative eval

Merge pull request !6131 from panfengfeng/fix_pynative_gpu_eval
pull/6131/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 234b7a6033

@ -15,6 +15,7 @@
*/
#include "runtime/device/gpu/gpu_kernel_runtime.h"
#include <algorithm>
#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"
@ -41,6 +42,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;
@ -322,6 +324,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);

Loading…
Cancel
Save