Browse Source

fix bug of unknown shape

tags/v1.2.0
wxl 4 years ago
parent
commit
260810227e
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      ge/graph/passes/flow_ctrl_pass.cc

+ 3
- 3
ge/graph/passes/flow_ctrl_pass.cc View File

@@ -33,11 +33,11 @@ Status FlowCtrlPass::Run(ComputeGraphPtr compute_graph) {
GE_CHECK_NOTNULL(compute_graph);

if (!PassUtils::IsNeedTrainIteFlowCtrl(compute_graph)) {
GELOGI("No need FlowCtrl for graph %u", compute_graph->GetGraphID());
GELOGI("No need FlowCtrl for graph %u.", compute_graph->GetGraphID());
return NOT_CHANGED;
}

GELOGI("FlowCtrl pass begin.graph is [%s]", compute_graph->GetName().c_str());
GELOGI("FlowCtrl pass begin.graph is [%s].", compute_graph->GetName().c_str());
bool graph_change = false;
// 1. Add FP/BP flow ctrl (big cycle)
for (auto &node : compute_graph->GetDirectNode()) {
@@ -86,7 +86,7 @@ Status FlowCtrlPass::Run(ComputeGraphPtr compute_graph) {
auto ret = GraphUtils::AddEdge(active_node->GetOutControlAnchor(),
assign_add_node_in_fpbp_loop_->GetInControlAnchor());
if (ret != GRAPH_SUCCESS) {
GELOGW("add control edge between iter_loop_node:%s and fpbp_loop_node:%s fail, may cause block",
GELOGW("add control edge between iter_loop_node:%s and fpbp_loop_node:%s fail, may cause block.",
active_node->GetName().c_str(), assign_add_node_in_fpbp_loop_->GetName().c_str());
}
}


Loading…
Cancel
Save