From f69636a817cf75fd32bb4cf68654249b0f1e0a97 Mon Sep 17 00:00:00 2001
From: "gengchao4@huawei.com" <gengchao4@huawei.com>
Date: Tue, 10 Nov 2020 21:57:39 +0800
Subject: [PATCH] add partition subgraph dump name suffix

---
 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..c76997d6 100644
--- a/ge/graph/partition/graph_partition.h
+++ b/ge/graph/partition/graph_partition.h
@@ -50,6 +50,7 @@ class Cluster {
   ~Cluster() = default;
 };
 using ClusterPtr = std::shared_ptr<Cluster>;
+using PartitionModeMap = std::map<Mode, std::string>;
 
 class GraphPartitioner {
  public:
@@ -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