Browse Source

Pre Merge pull request !101 from gengchao/master

pull/101/MERGE
gengchao Gitee 5 years ago
parent
commit
1b4709c157
2 changed files with 6 additions and 1 deletions
  1. +1
    -0
      src/common/graph/utils/graph_utils.cc
  2. +5
    -1
      src/common/graph/utils/tuning_utils.cc

+ 1
- 0
src/common/graph/utils/graph_utils.cc View File

@@ -60,6 +60,7 @@ const int32_t kBaseOfIntegerValue = 10;
const char *const kDumpGeGraph = "DUMP_GE_GRAPH"; const char *const kDumpGeGraph = "DUMP_GE_GRAPH";
const int kDumpGraphIndexWidth = 5; const int kDumpGraphIndexWidth = 5;
#endif #endif
const char *const kDumpGraphPath = "DUMP_GRAPH_PATH";
const char *const kDumpGraphLevel = "DUMP_GRAPH_LEVEL"; const char *const kDumpGraphLevel = "DUMP_GRAPH_LEVEL";
const char *const kDumpStrBuild = "Build"; const char *const kDumpStrBuild = "Build";
const char *const kDumpStrPartition = "partition"; const char *const kDumpStrPartition = "partition";


+ 5
- 1
src/common/graph/utils/tuning_utils.cc View File

@@ -116,6 +116,10 @@ graphStatus TuningUtils::ConvertGraphToFile(std::vector<ComputeGraphPtr> tuning_
// +---------------+ // +---------------+
graphStatus TuningUtils::MakeExeGraph(ComputeGraphPtr &exe_graph, const HelpInfo &help_info) { graphStatus TuningUtils::MakeExeGraph(ComputeGraphPtr &exe_graph, const HelpInfo &help_info) {
GE_CHECK_NOTNULL(exe_graph); GE_CHECK_NOTNULL(exe_graph);
// clear graph id
GELOGI("TUU:clear [%s] session_graph_id %s",
exe_graph->GetName().c_str(),
(AttrUtils::SetStr(*exe_graph, ATTR_NAME_SESSION_GRAPH_ID, "") ? "success" : "not success"));
// if not make exe, just dump and return // if not make exe, just dump and return
if (!help_info.exe_flag) { if (!help_info.exe_flag) {
DumpGraphToPath(exe_graph, help_info.index, help_info.is_tuning_graph, help_info.path); DumpGraphToPath(exe_graph, help_info.index, help_info.is_tuning_graph, help_info.path);
@@ -681,4 +685,4 @@ graphStatus TuningUtils::GetInAndOutAnchorPair(NodePtr &data_node, NodePtr &out_
return SUCCESS; return SUCCESS;
} }


} // namespace ge
} // namespace ge

Loading…
Cancel
Save