Browse Source

mode partition subgrap dump graph name

pull/219/head
gengchao4@huawei.com 4 years ago
parent
commit
9d1b59bb06
4 changed files with 5 additions and 3 deletions
  1. +1
    -1
      ge/graph/partition/graph_partition.cc
  2. +2
    -0
      ge/graph/partition/graph_partition.h
  3. +1
    -1
      metadef
  4. +1
    -1
      parser

+ 1
- 1
ge/graph/partition/graph_partition.cc View File

@@ -586,7 +586,7 @@ Status ge::GraphPartitioner::AddPartitionsToGraphNode(vector<ge::SubGraphInfoPtr
return FAILED; return FAILED;
} }
auto &engine_name = graph_info_.partitions_.at(sub_graph); auto &engine_name = graph_info_.partitions_.at(sub_graph);
GE_DUMP(sub_graph, sub_graph->GetName());
GE_DUMP(sub_graph, sub_graph->GetName() + "_" + mode2str[graph_info_.mode_]);
if (!session_graph_id.empty()) { if (!session_graph_id.empty()) {
GE_IF_BOOL_EXEC(!AttrUtils::SetStr(sub_graph, ATTR_NAME_SESSION_GRAPH_ID, session_graph_id), GE_IF_BOOL_EXEC(!AttrUtils::SetStr(sub_graph, ATTR_NAME_SESSION_GRAPH_ID, session_graph_id),
GELOGW("SetStr ATTR_NAME_SESSION_GRAPH_ID failed");) GELOGW("SetStr ATTR_NAME_SESSION_GRAPH_ID failed");)


+ 2
- 0
ge/graph/partition/graph_partition.h View File

@@ -57,6 +57,8 @@ class GraphPartitioner {
/// MergeAfterSubGraphOptimization() can only be called in Merge mode. /// MergeAfterSubGraphOptimization() can only be called in Merge mode.
/// After Partition(), change to Merge mode. After MergeAfterSubGraphOptimization(), change to Partition mode /// After Partition(), change to Merge mode. After MergeAfterSubGraphOptimization(), change to Partition mode
enum Mode { kPartitioning, kSecondPartitioning, kMerging }; enum Mode { kPartitioning, kSecondPartitioning, kMerging };
std::map<Mode, std::string> mode2str =
{{kPartitioning, "kPartitioning"}, {kSecondPartitioning, "kSecondPartitioning"}, {kMerging, "kMerging"}};
GraphPartitioner() : partition_times_(0){}; GraphPartitioner() : partition_times_(0){};
~GraphPartitioner() = default; ~GraphPartitioner() = default;




+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit c3543bc707bdfd60ef4f52b6dffcdfefdf3b8329
Subproject commit 4c1f3071b2a66295f6aab82a6a698ef47572cb64

+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit d2dcd37ead34bb95a81ccb08b62e79f8f4b102b5
Subproject commit db4e6070bb2cec01cead264a44ceae07e7f3048e

Loading…
Cancel
Save