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(vectorGetName()); + 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..8a8ed65c 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; 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