From d207cc443e156101ef415a15106cf82298fdf194 Mon Sep 17 00:00:00 2001
From: Parastoo Ashtari <parastoo.ashtari@huawei.com>
Date: Tue, 5 Apr 2022 17:00:59 -0400
Subject: [PATCH] fix EnableDebugger called so many times for dump

---
 mindspore/ccsrc/debug/debugger/debugger.cc | 6 ++++--
 mindspore/ccsrc/debug/debugger/debugger.h  | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/debug/debugger/debugger.cc b/mindspore/ccsrc/debug/debugger/debugger.cc
index 8c70adc5fb3..90fe1404e09 100644
--- a/mindspore/ccsrc/debug/debugger/debugger.cc
+++ b/mindspore/ccsrc/debug/debugger/debugger.cc
@@ -86,6 +86,7 @@ Debugger::Debugger()
       enable_heartbeat_(false),
       not_dataset_graph_sum_(0),
       ascend_kernel_by_kernel_(false),
+      enable_debugger_called_(false),
       version_("") {
   CheckDebuggerEnabledParam();
   auto ms_context = MsContext::GetInstance();
@@ -143,6 +144,7 @@ void Debugger::EnableDebugger() {
   grpc_client_ = nullptr;
   debug_services_ = nullptr;
   heartbeat_thread_ = nullptr;
+  enable_debugger_called_ = true;
 
   // see if dump using debugger backend is enabled
   bool dump_enabled = CheckDebuggerDumpEnabled();
@@ -440,7 +442,7 @@ void Debugger::PreExecute(const KernelGraphPtr &graph_ptr) {
  */
 void Debugger::SendMultiGraphsAndClear(const KernelGraphPtr &graph_ptr) {
   // only try to enable debugger if they are not all dataset graphs
-  if (!debugger_enabled_) {
+  if (!enable_debugger_called_) {
     EnableDebugger();
   }
   if (debugger_enabled_) {
@@ -710,7 +712,7 @@ void Debugger::CheckGraphPtr(const KernelGraphPtr &graph_ptr) {
     graph_ptr_ = graph_ptr;
     if (!is_dataset_graph_) {
       // only try to enable debugger if it is not a dataset graph
-      if (!debugger_enabled_) {
+      if (!enable_debugger_called_) {
         EnableDebugger();
       }
       if (debugger_enabled_) {
diff --git a/mindspore/ccsrc/debug/debugger/debugger.h b/mindspore/ccsrc/debug/debugger/debugger.h
index f9d43616980..b6a06c161f4 100644
--- a/mindspore/ccsrc/debug/debugger/debugger.h
+++ b/mindspore/ccsrc/debug/debugger/debugger.h
@@ -337,6 +337,7 @@ class BACKEND_EXPORT Debugger : public std::enable_shared_from_this<Debugger> {
   uint32_t not_dataset_graph_sum_;
   std::list<uint32_t> rungraph_id_list_;
   bool ascend_kernel_by_kernel_;
+  bool enable_debugger_called_;
   std::string version_;
 };