add longrunning for runop

pull/9256/head
kswang 4 years ago
parent 4bd9460e45
commit 0ad5357ce7

@ -357,6 +357,7 @@ void Executor::RunOp(const SessionPtr &session, OpRunInfo *op_run_info, const Gr
tensor->Wait();
}
}
mindspore::ScopedLongRunning long_running;
SyncRunTask(task);
*outputs = task->outputs_;
}

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

Loading…
Cancel
Save