Browse Source

display model info

pull/769/head
wangwenhua1@huawei.com 4 years ago
parent
commit
41282db089
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      ge/common/helper/model_helper.cc

+ 3
- 3
ge/common/helper/model_helper.cc View File

@@ -103,8 +103,8 @@ Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model) {


std::shared_ptr<ModelTaskDef> model_task_def = ge_model->GetModelTaskDefPtr(); std::shared_ptr<ModelTaskDef> model_task_def = ge_model->GetModelTaskDefPtr();
if (model_task_def == nullptr) { if (model_task_def == nullptr) {
GELOGE(ACL_ERROR_GE_MEMORY_ALLOCATION, "Create model task def ptr failed");
return ACL_ERROR_GE_MEMORY_ALLOCATION;
GELOGE(FAILED, "Create model task def ptr failed");
return FAILED;
} }
size_t partition_task_size = model_task_def->ByteSizeLong(); size_t partition_task_size = model_task_def->ByteSizeLong();
GELOGD("SaveSizeToModelDef task_info_size is %zu", partition_task_size); GELOGD("SaveSizeToModelDef task_info_size is %zu", partition_task_size);
@@ -233,7 +233,7 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY Status ModelHelper::SaveToOmMod
string model_name = reinterpret_cast<char *>(model_header.name); string model_name = reinterpret_cast<char *>(model_header.name);
GELOGI("Model name save:%s", model_name.c_str()); GELOGI("Model name save:%s", model_name.c_str());


Status ret = om_file_save_helper->SaveModel(save_param, output_file.c_str(), model, is_offline_);
ret = om_file_save_helper->SaveModel(save_param, output_file.c_str(), model, is_offline_);
if (ret != SUCCESS) { if (ret != SUCCESS) {
GELOGE(FAILED, "OmFileSaveHelper SaveModel return fail."); GELOGE(FAILED, "OmFileSaveHelper SaveModel return fail.");
return FAILED; return FAILED;


Loading…
Cancel
Save