From 16c2f091b742899b42bad90aaff2aeefb05db5e0 Mon Sep 17 00:00:00 2001 From: unknown Date: Wed, 3 Feb 2021 20:58:14 +0800 Subject: [PATCH] Check dynamic support. --- ge/generator/ge_generator.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 9d9fc23c..69ee2bfa 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -632,8 +632,8 @@ namespace { } auto op_desc = node->GetOpDesc(); GE_CHECK_NOTNULL(op_desc); - (void)AttrUtils::GetBool(op_desc, kAttrSupportDynamicShape, is_dynamic); - if (!is_dynamic) { + (void)AttrUtils::GetBool(op_desc, kAttrSupportDynamicShape, support_dynamic); + if (!support_dynamic) { GELOGW("Node[%s] doesn't support dynamic shape.", node->GetName().c_str()); (void)AttrUtils::SetBool(ge_model, kAttrSupportDynamicShape, false); return SUCCESS; @@ -717,7 +717,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector &in } const ComputeGraphPtr root_graph = ge_root_model->GetRootGraph(); GeModelPtr &ge_model = name_to_ge_model.begin()->second; - E_CHK_STATUS_RET_NOLOG(CheckDynamicSupport(ge_model, root_graph)); + GE_CHK_STATUS_RET_NOLOG(CheckDynamicSupport(ge_model, root_graph)); GELOGD("The opType in op_desc_tmp is [%s]", op_desc_tmp->GetType().c_str()); bool all_shape = false;