| @@ -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; | ||||