From 0c16b866fee3610ea1f6b8bb54bb7df59582dc0b Mon Sep 17 00:00:00 2001 From: wangyue01 Date: Tue, 3 Nov 2020 19:38:20 +0800 Subject: [PATCH] Save graph before remove nop nodes --- mindspore/ccsrc/backend/session/ascend_session.cc | 8 ++++++++ mindspore/ccsrc/backend/session/gpu_session.cc | 5 +++++ 2 files changed, 13 insertions(+) diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc index b43777c4a6..ccbb4da45c 100644 --- a/mindspore/ccsrc/backend/session/ascend_session.cc +++ b/mindspore/ccsrc/backend/session/ascend_session.cc @@ -170,6 +170,14 @@ GraphId AscendSession::CompileGraphImpl(NotNull func_graph) { InsertMakeTupleForOutput(NOT_NULL(root_graph)); // root root_graph valiate,include genearte execute order and so on RootGraphExecutorValidate(NOT_NULL(root_graph)); + // dump graph before remove nop nodes + auto context_ptr = MsContext::GetInstance(); + MS_EXCEPTION_IF_NULL(context_ptr); + bool save_graphs = context_ptr->get_param(MS_CTX_SAVE_GRAPHS_FLAG); + if (save_graphs) { + DumpIRProto(root_graph, "before_removeNop_" + std::to_string(graph_sum_)); + } + // adjust kernel AdjustKernel(root_graph); #if ENABLE_CPU && ENABLE_D diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index 779b1d3d26..7ee81a0046 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -302,6 +302,11 @@ GraphId GPUSession::CompileGraphImpl(const AnfNodePtrList &lst, const AnfNodePtr StartKernelRT(); // Assign CUDA streams AssignStream(graph); + // Dump .pb graph before remove nop nodes + if (save_graphs) { + DumpIRProto(graph, "before_removeNop_" + std::to_string(graph_id)); + } + // Hide NopOp from execution graph opt::HideNopNode(graph.get()); // Build kernel if node is cnode