From 9ac8c3e181df5626d03b7ec32ce1d7adbce9b69c Mon Sep 17 00:00:00 2001 From: lichun Date: Sat, 13 Mar 2021 10:21:47 +0800 Subject: [PATCH] dynamic shape inference support --- ge/hybrid/model/node_item.cc | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/ge/hybrid/model/node_item.cc b/ge/hybrid/model/node_item.cc index 100530fc..d2aa4a1f 100644 --- a/ge/hybrid/model/node_item.cc +++ b/ge/hybrid/model/node_item.cc @@ -152,11 +152,13 @@ Status NodeItem::InitInputsAndOutputs() { if (AttrUtils::GetInt(node->GetOwnerComputeGraph(), ::ge::ATTR_STAGE_LEVEL, group)) { GELOGD("[%s] Got stage level from parent graph = %d", op_desc->GetName().c_str(), group); } else { - auto parent_node = node->GetOwnerComputeGraph()->GetParentNode(); - if ((parent_node != nullptr) && (AttrUtils::GetInt(parent_node->GetOpDesc(), ::ge::ATTR_STAGE_LEVEL, group))) { - GELOGD("[%s] Got stage level from parent node = %d", op_desc->GetName().c_str(), group); - } else { - GELOGD("[%s] Node do not set stage level", op_desc->GetName().c_str()); + if (node->GetOwnerComputeGraph() != nullptr) { + auto parent_node = node->GetOwnerComputeGraph()->GetParentNode(); + if ((parent_node != nullptr) && (AttrUtils::GetInt(parent_node->GetOpDesc(), ::ge::ATTR_STAGE_LEVEL, group))) { + GELOGD("[%s] Got stage level from parent node = %d", op_desc->GetName().c_str(), group); + } else { + GELOGD("[%s] Node do not set stage level", op_desc->GetName().c_str()); + } } } }