Browse Source

modify single op dynamic shape

pull/1633/head
wqtshg 4 years ago
parent
commit
0ccd196b50
4 changed files with 7 additions and 8 deletions
  1. +4
    -6
      ge/graph/load/model_manager/model_manager.cc
  2. +1
    -0
      ge/model/ge_root_model.cc
  3. +1
    -1
      metadef
  4. +1
    -1
      parser

+ 4
- 6
ge/graph/load/model_manager/model_manager.cc View File

@@ -1140,13 +1140,11 @@ Status ModelManager::LoadModelOffline(uint32_t &model_id, const ModelData &model
return ret; return ret;
} }


if (model_helper.GetModelType()) {
bool is_shape_unknown = false;
GE_CHK_STATUS_RET(model_helper.GetGeRootModel()->CheckIsUnknownShape(is_shape_unknown),
"CheckIsUnknownShape failed, model id:%u", model_id);
if (is_shape_unknown || GetContext().GetHostExecFlag()) {
bool is_shape_unknown = false;
GE_CHK_STATUS_RET(model_helper.GetGeRootModel()->CheckIsUnknownShape(is_shape_unknown),
"CheckIsUnknownShape failed, model id:%u", model_id);
if (is_shape_unknown || GetContext().GetHostExecFlag()) {
return DoLoadHybridModelOnline(model_id, model.om_name, model_helper.GetGeRootModel(), listener); return DoLoadHybridModelOnline(model_id, model.om_name, model_helper.GetGeRootModel(), listener);
}
} }


do { do {


+ 1
- 0
ge/model/ge_root_model.cc View File

@@ -27,6 +27,7 @@ Status GeRootModel::CheckIsUnknownShape(bool &is_dynamic_shape) {
} }
is_dynamic_shape = false; is_dynamic_shape = false;
(void)AttrUtils::GetBool(root_graph_, ATTR_NAME_DYNAMIC_SHAPE_PARTITIONED, is_dynamic_shape); (void)AttrUtils::GetBool(root_graph_, ATTR_NAME_DYNAMIC_SHAPE_PARTITIONED, is_dynamic_shape);
is_dynamic_shape = is_dynamic_shape || root_graph_->GetGraphUnknownFlag();
return SUCCESS; return SUCCESS;
} }
} // namespace ge } // namespace ge

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 68474443bd6966eade3e32d6dfa2cc62f5872d2c
Subproject commit 0facaa5af36b64c9d39603ed419191d21832df8a

+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit 7773435b776fb37231abcef2bbcf972814b01dd1
Subproject commit 055cc4f21aac44b96ef2f951d9e7f89f7042d77a

Loading…
Cancel
Save