From 1c5ffebc07730e476dbe4117e9e1c00721e6a5f9 Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Tue, 16 Mar 2021 14:44:18 +0800
Subject: [PATCH] fix bug of dynamic shape load error

---
 ge/graph/load/model_manager/model_manager.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ge/graph/load/model_manager/model_manager.cc b/ge/graph/load/model_manager/model_manager.cc
index 97ad0054..3b324a1a 100755
--- a/ge/graph/load/model_manager/model_manager.cc
+++ b/ge/graph/load/model_manager/model_manager.cc
@@ -299,9 +299,9 @@ Status ModelManager::LoadModelOnline(uint32_t &model_id, const shared_ptr<ge::Ge
   }
   auto name_to_model = ge_root_model->GetSubgraphInstanceNameToModel();
   string model_name = "";
-  bool is_shape_unknown = ge_root_model->GetRootGraph()->GetGraphUnknownFlag();
-  // if multi subgraph is known, do hybrid load process
-  if (is_shape_unknown || GetContext().GetHostExecFlag() || (name_to_model.size() > 1)) {
+  bool is_shape_unknown = false;
+  (void)AttrUtils::GetBool(ge_root_model->GetRootGraph(), ATTR_NAME_DYNAMIC_SHAPE_PARTITIONED, is_shape_unknown);
+  if (is_shape_unknown || GetContext().GetHostExecFlag()) {
     return DoLoadHybridModelOnline(model_id, model_name, ge_root_model, listener);
   }