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_ptrGetSubgraphInstanceNameToModel(); 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); }