Browse Source

bugfix for check offset

tags/v1.3.0
gengchao4@huawei.com 4 years ago
parent
commit
b2c9de02f9
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      ge/graph/build/memory/graph_mem_assigner.cc

+ 5
- 0
ge/graph/build/memory/graph_mem_assigner.cc View File

@@ -1258,6 +1258,11 @@ Status GraphMemoryAssigner::CheckOffset() {
}

ge::Status GraphMemoryAssigner::CheckRefNodeOffset(const NodePtr &node) {
GE_CHECK_NOTNULL(node);
// data and netoutput no need check because only data's output or netoutput's input is used
if (node->GetType() == DATA || node->GetType() == NETOUTPUT) {
return ge::SUCCESS;
}
std::map<int32_t, int32_t> out2ins;
GE_CHK_STATUS_RET(TryGetNodeRefIndexes(node, out2ins), "[Get][RefIndexes]fail for node: %s", node->GetName().c_str());
auto opdesc = node->GetOpDesc();


Loading…
Cancel
Save