xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!3397 GPU dump - input bins lag behind by 1 iteration 

Merge pull request !3397 from john_tzanakakis/master_ms1
This commit is contained in:
mindspore-ci-bot 2020-07-24 21:58:09 +08:00 committed by Gitee
parent 514c5d98b2 96744f087e
commit 252c6aa788
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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