m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix bug of unknown shape

Browse Source
pull/1213/head
wxl 4 years ago
parent b9d4a05c21
commit 260810227e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File

6
ge/graph/passes/flow_ctrl_pass.cc
Unescape View File

@ -33,11 +33,11 @@ Status FlowCtrlPass::Run(ComputeGraphPtr compute_graph) {
GE_CHECK_NOTNULL(compute_graph); GE_CHECK_NOTNULL(compute_graph);
if (!PassUtils::IsNeedTrainIteFlowCtrl(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; 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; bool graph_change = false;
// 1. Add FP/BP flow ctrl (big cycle) // 1. Add FP/BP flow ctrl (big cycle)
for (auto &node : compute_graph->GetDirectNode()) { for (auto &node : compute_graph->GetDirectNode()) {
@ -86,7 +86,7 @@ Status FlowCtrlPass::Run(ComputeGraphPtr compute_graph) {
auto ret = GraphUtils::AddEdge(active_node->GetOutControlAnchor(), auto ret = GraphUtils::AddEdge(active_node->GetOutControlAnchor(),
assign_add_node_in_fpbp_loop_->GetInControlAnchor()); assign_add_node_in_fpbp_loop_->GetInControlAnchor());
if (ret != GRAPH_SUCCESS) { 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()); active_node->GetName().c_str(), assign_add_node_in_fpbp_loop_->GetName().c_str());
} }
} }

Write Preview
Loading…
Cancel
Save