|
|
|
@@ -16,7 +16,6 @@ |
|
|
|
#include "runtime/device/gpu/gpu_kernel_runtime.h" |
|
|
|
#include <algorithm> |
|
|
|
#include <map> |
|
|
|
#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" |
|
|
|
@@ -45,7 +44,6 @@ |
|
|
|
namespace mindspore { |
|
|
|
namespace device { |
|
|
|
namespace gpu { |
|
|
|
namespace py = pybind11; |
|
|
|
using mindspore::device::memswap::MemSwapInfoSet; |
|
|
|
using mindspore::device::memswap::MemSwapManager; |
|
|
|
using mindspore::device::memswap::SwapKind; |
|
|
|
@@ -362,7 +360,6 @@ bool GPUKernelRuntime::Run(session::KernelGraph *graph, bool is_task_sink) { |
|
|
|
|
|
|
|
ret = RunOneStep(graph); |
|
|
|
} else { |
|
|
|
py::gil_scoped_release gil_release; |
|
|
|
ret = LaunchKernel(graph); |
|
|
|
} |
|
|
|
(void)gettimeofday(&end_time, nullptr); |
|
|
|
|