Browse Source

GPU dump - input bins lag behind by 1 iteration

tags/v0.7.0-beta
John Tzanakakis 5 years ago
parent
commit
96744f087e
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc

+ 4
- 4
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc View File

@@ -616,10 +616,6 @@ bool GPUKernelRuntime::LaunchKernelDynamic(const session::KernelGraph *graph, De
#ifdef ENABLE_DEBUGGER
bool dump_enabled = GPUKernelRuntime::DumpDataEnabledIteration();
if (!mock) {
// collect weights and bias
LoadParameters(graph, debugger, dump_enabled);
}
#endif
auto &kernels = graph->execution_order();
int exec_order = 1;
@@ -665,6 +661,10 @@ bool GPUKernelRuntime::LaunchKernelDynamic(const session::KernelGraph *graph, De
}
}
if (!mock) {
#ifdef ENABLE_DEBUGGER
// collect weights and bias
LoadParameters(graph, debugger, dump_enabled);
#endif
CHECK_OP_RET_WITH_EXCEPT(SyncStream(), "SyncStream failed.");
}
ClearSwapInfo(mock);


Loading…
Cancel
Save