huawei
/
mindspore2022
0
0
Fork 53
Code Issues Pull Requests 29 Projects Releases Wiki Activity

fix EnableDebugger called so many times for dump

This commit is contained in:
Parastoo Ashtari 2022-04-05 17:00:59 -04:00
parent e84f16ac7d
commit d207cc443e
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/debug/debugger/debugger.cc
View File

@ -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_) {

1
mindspore/ccsrc/debug/debugger/debugger.h
View File

@ -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_;
};