Browse Source

fix bug of dynamic shape load error

tags/v1.3.0
wxl 4 years ago
parent
commit
f8acefb43d
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      ge/graph/passes/identity_pass.cc
  2. +1
    -1
      ge/graph/passes/prune_pass.cc

+ 2
- 2
ge/graph/passes/identity_pass.cc View File

@@ -54,10 +54,10 @@ Status CheckIdentityUsable(const NodePtr &node, bool &usable) {
}

GE_CHK_STATUS_RET(GetOriginalType(in_node, node_type),
"Failed to get node type from node %s.", node->GetName().c_str());
"Failed to get node type from node %s", node->GetName().c_str());
bool need_skip = (node_type != SWITCH) && (node_type != REFSWITCH) && (node_type != SWITCHN);
if (need_skip) {
GELOGD("skip identity %s connected to switch.", node->GetName().c_str());
GELOGD("skip identity %s connected to switch", node->GetName().c_str());
break;
}
GE_CHECK_NOTNULL(node->GetOutControlAnchor());


+ 1
- 1
ge/graph/passes/prune_pass.cc View File

@@ -42,7 +42,7 @@ Status PrunePass::Run(ge::ComputeGraphPtr graph) {
}
}
if (out_nodes.empty()) {
GELOGW("graph [%s] does not contain NETOUTPUT type node, no return value. Do nothing!", graph->GetName().c_str());
GELOGW("graph [%s] does not contain NETOUTPUT type node,no return value. Do nothing!", graph->GetName().c_str());
return ge::SUCCESS;
}



Loading…
Cancel
Save