Browse Source

revert for --161 mstune bugfix for build_step of after merge

tags/v1.1.0
gengchao4@huawei.com 3 years ago
parent
commit
42b8d4f2e5
2 changed files with 11 additions and 6 deletions
  1. +10
    -4
      ge/graph/manager/graph_manager.cc
  2. +1
    -2
      ge/graph/optimize/graph_optimize.cc

+ 10
- 4
ge/graph/manager/graph_manager.cc View File

@@ -693,10 +693,16 @@ Status GraphManager::PreRun(const GraphNodePtr &graph_node, const std::vector<Ge
}
}

ret = PreRunOptimizeSubGraph(graph_node, compute_graph, session_id);
if (ret != SUCCESS) {
GELOGE(ret, "Run PreRunOptimizeSubGraph failed for graph:%s.", compute_graph->GetName().c_str());
return ret;

// BUILD_MODE_TUNING with BUILD_STEP_AFTER_MERGE no need PreRunOptimizeSubGraph.
bool run_optimize_subgraph =
!((options_.build_mode == BUILD_MODE_TUNING) && (options_.build_step == BUILD_STEP_AFTER_MERGE));
if (run_optimize_subgraph) {
Status ret = PreRunOptimizeSubGraph(graph_node, compute_graph, session_id);
if (ret != SUCCESS) {
GELOGE(ret, "Run PreRunOptimizeSubGraph failed for graph:%s.", compute_graph->GetName().c_str());
return ret;
}
}

/// 1. BUILD_MODE_TUNING with BUILD_STEP_BEFORE_UB_MATCH no need PreRunAfterOptimizeSubGraph;


+ 1
- 2
ge/graph/optimize/graph_optimize.cc View File

@@ -101,8 +101,7 @@ Status GraphOptimize::OptimizeSubGraph(ComputeGraphPtr &compute_graph, const std
return SUCCESS;
}

if (build_mode_ == BUILD_MODE_TUNING &&
(build_step_ == BUILD_STEP_AFTER_UB_MATCH || build_step_ == BUILD_STEP_AFTER_MERGE)) {
if (build_mode_ == BUILD_MODE_TUNING && build_step_ == BUILD_STEP_AFTER_UB_MATCH) {
for (auto iter = graph_optimizer.begin(); iter != graph_optimizer.end(); ++iter) {
Status ret = (*iter)->OptimizeFusedGraphAfterGraphSlice(*(compute_graph));
if (ret != SUCCESS) {


Loading…
Cancel
Save