From cbbffbedef053fcc747f5c1f891623a24a61a7a7 Mon Sep 17 00:00:00 2001 From: Islam Amin Date: Thu, 11 Mar 2021 12:02:52 -0500 Subject: [PATCH] fix gpu dump naming --- mindspore/ccsrc/backend/session/gpu_session.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index c134cc0569..3a3354f0a8 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -380,10 +380,11 @@ GraphId GPUSession::CompileGraphImpl(KernelGraphPtr graph) { DumpIRProto(graph, "after_opt_" + std::to_string(graph->graph_id())); } if (json_parser.e2e_dump_enabled()) { + std::string final_graph = "trace_code_graph_" + std::to_string(graph->graph_id()); std::string root_dir = json_parser.path() + "/" + json_parser.net_name() + "/device_" + std::to_string(device_id); std::string target_dir = root_dir + "/graphs"; - std::string ir_file_path = target_dir + "/" + "after_opt_" + std::to_string(graph->graph_id()) + ".ir"; - DumpIRProtoWithSrcInfo(graph, "after_opt_" + std::to_string(graph->graph_id()), target_dir, kDebugWholeStack); + std::string ir_file_path = target_dir + "/" + "ms_output_" + final_graph + ".ir"; + DumpIRProtoWithSrcInfo(graph, final_graph, target_dir, kDebugWholeStack); DumpIR("trace_code_graph", graph, true, kWholeStack, ir_file_path); DumpGraphExeOrder("ms_execution_order_graph_" + std::to_string(graph->graph_id()) + ".csv", root_dir, graph->execution_order());