From 29662329961e57842e0fb8469c2fe090c5efb692 Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Wed, 17 Mar 2021 14:09:48 +0800
Subject: [PATCH] fix bug of dynamic shape load error

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

diff --git a/ge/graph/load/model_manager/model_manager.cc b/ge/graph/load/model_manager/model_manager.cc
index 3b324a1a..aa2de7e6 100755
--- a/ge/graph/load/model_manager/model_manager.cc
+++ b/ge/graph/load/model_manager/model_manager.cc
@@ -299,9 +299,10 @@ 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 = false;
+  bool is_shape_unknown = ge_root_model->GetRootGraph()->GetGraphUnknownFlag();
+  bool is_dsp_partitioned_graph = false;
   (void)AttrUtils::GetBool(ge_root_model->GetRootGraph(), ATTR_NAME_DYNAMIC_SHAPE_PARTITIONED, is_shape_unknown);
-  if (is_shape_unknown || GetContext().GetHostExecFlag()) {
+  if (is_shape_unknown || is_dsp_partitioned_graph || GetContext().GetHostExecFlag()) {
     return DoLoadHybridModelOnline(model_id, model_name, ge_root_model, listener);
   }