From c5124091500da5643a768e88dee6f39192fc2bc3 Mon Sep 17 00:00:00 2001
From: "gengchao4@huawei.com" <gengchao4@huawei.com>
Date: Wed, 11 Nov 2020 10:45:25 +0800
Subject: [PATCH] modify subgraph dump name

---
 ge/graph/partition/graph_partition.cc | 2 +-
 ge/graph/partition/graph_partition.h  | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/ge/graph/partition/graph_partition.cc b/ge/graph/partition/graph_partition.cc
index c8e8cda3..e1b645b3 100755
--- a/ge/graph/partition/graph_partition.cc
+++ b/ge/graph/partition/graph_partition.cc
@@ -586,7 +586,7 @@ Status ge::GraphPartitioner::AddPartitionsToGraphNode(vector<ge::SubGraphInfoPtr
       return FAILED;
     }
     auto &engine_name = graph_info_.partitions_.at(sub_graph);
-    GE_DUMP(sub_graph, sub_graph->GetName());
+    GE_DUMP(sub_graph, sub_graph->GetName()  + "_" + mode_2_str_[graph_info_.mode_]);
     if (!session_graph_id.empty()) {
       GE_IF_BOOL_EXEC(!AttrUtils::SetStr(sub_graph, ATTR_NAME_SESSION_GRAPH_ID, session_graph_id),
                       GELOGW("SetStr ATTR_NAME_SESSION_GRAPH_ID failed");)
diff --git a/ge/graph/partition/graph_partition.h b/ge/graph/partition/graph_partition.h
index 703a1570..003c3a06 100644
--- a/ge/graph/partition/graph_partition.h
+++ b/ge/graph/partition/graph_partition.h
@@ -57,6 +57,7 @@ class GraphPartitioner {
   /// MergeAfterSubGraphOptimization() can only be called in Merge mode.
   /// After Partition(), change to Merge mode. After MergeAfterSubGraphOptimization(), change to Partition mode
   enum Mode { kPartitioning, kSecondPartitioning, kMerging };
+  using PartitionModeMap = std::map<Mode, std::string>;
   GraphPartitioner() : partition_times_(0){};
   ~GraphPartitioner() = default;
 
@@ -176,6 +177,8 @@ class GraphPartitioner {
   Graph2InputNodesSubGraphInfo graph_2_input_subgraph_;
   GraphPartitionInfo graph_info_;
   uint32_t partition_times_;  // times of call partition
+  PartitionModeMap mode_2_string_ = {{kPartitioning, "Partitioning"},
+    {kSecondPartitioning, "SecondPartitioning"}, {kMerging, "Merging"}};
   friend class GraphManager;
 };
 }  // namespace ge