diff --git a/ge/graph/partition/graph_partition.cc b/ge/graph/partition/graph_partition.cc index c8e8cda3..34da5777 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() + "_" + mode2str[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..9c3626f0 100644 --- a/ge/graph/partition/graph_partition.h +++ b/ge/graph/partition/graph_partition.h @@ -57,6 +57,8 @@ 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 }; + std::map mode2str = + {{kPartitioning, "kPartitioning"}, {kSecondPartitioning, "kSecondPartitioning"}, {kMerging, "kMerging"}}; GraphPartitioner() : partition_times_(0){}; ~GraphPartitioner() = default; diff --git a/metadef b/metadef index c3543bc7..4c1f3071 160000 --- a/metadef +++ b/metadef @@ -1 +1 @@ -Subproject commit c3543bc707bdfd60ef4f52b6dffcdfefdf3b8329 +Subproject commit 4c1f3071b2a66295f6aab82a6a698ef47572cb64 diff --git a/parser b/parser index d2dcd37e..db4e6070 160000 --- a/parser +++ b/parser @@ -1 +1 @@ -Subproject commit d2dcd37ead34bb95a81ccb08b62e79f8f4b102b5 +Subproject commit db4e6070bb2cec01cead264a44ceae07e7f3048e