Fix the unchanged iteration id on cpu

2021-07-09 15:09:57 +08:00 · 2021-07-09 15:09:57 +08:00 · 97005570a3
parent 3a408218ec
commit 97005570a3
1 changed files with 5 additions and 0 deletions
--- a/mindspore/ccsrc/debug/debugger/debugger.cc
+++ b/mindspore/ccsrc/debug/debugger/debugger.cc
@ -442,6 +442,11 @@ void Debugger::DumpInGraphCompiler(const KernelGraphPtr &kernel_graph) {
 }

 void Debugger::PostExecuteGraphDebugger() {
+  // On CPU, update dump iteration， Parameters and consts are not dumped here
+  if (device_target_ == kCPUDevice) {
+    DumpJsonParser::GetInstance().UpdateDumpIter();
+    return;
+  }
  // Only GPU is supported for MindRTBackend
  if (device_target_ != kGPUDevice) {
    return;