|
|
|
@ -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);
|
|
|
|
|