|
|
|
@@ -560,7 +560,10 @@ static Status SetModelNameForDump(GeRootModelPtr ge_root_model) { |
|
|
|
bool is_unknown_shape = false; |
|
|
|
Status ret = ge_root_model->CheckIsUnknownShape(is_unknown_shape); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
GELOGE(FAILED, "Check root model is unknown shape failed."); |
|
|
|
GELOGE(FAILED, "[Check][IsUnknownShape]Check root model is unknown shape failed, model id:%zu", |
|
|
|
ge_root_model->GetModelId()); |
|
|
|
REPORT_CALL_ERROR("E19999", "Check root model is unknown shape failed, model id:%zu", |
|
|
|
ge_root_model->GetModelId()); |
|
|
|
return FAILED; |
|
|
|
} |
|
|
|
GeModelPtr model_root = nullptr; |
|
|
|
@@ -578,12 +581,15 @@ static Status SetModelNameForDump(GeRootModelPtr ge_root_model) { |
|
|
|
model_name); |
|
|
|
if (name_ret != SUCCESS) { |
|
|
|
ErrorManager::GetInstance().ATCReportErrMessage("E10000", {"parameter"}, {"output"}); |
|
|
|
GELOGE(FAILED, "Get model_name failed. Param --output is invalid"); |
|
|
|
GELOGE(FAILED, "[Check][GetModelNameStep]Get model_name failed. Param --output is invalid,", |
|
|
|
"root graph name: %s", ge_root_model->GetRootGraph()->GetName().c_str()); |
|
|
|
REPORT_CALL_ERROR("E19999", "Get model_name failed. Param --output is invalid,", |
|
|
|
"root graph name: %s", ge_root_model->GetRootGraph()->GetName().c_str()); |
|
|
|
return PARAM_INVALID; |
|
|
|
} |
|
|
|
map<string, GeModelPtr> name_to_ge_model = ge_root_model->GetSubgraphInstanceNameToModel(); |
|
|
|
GeModelPtr &ge_model = name_to_ge_model[ge_root_model->GetRootGraph()->GetName()]; |
|
|
|
GE_RETURN_WITH_LOG_IF_FALSE(ge_model != nullptr, "ge_model cannot be null."); |
|
|
|
GE_CHECK_NOTNULL(ge_model); |
|
|
|
ge_model->SetName(model_name); |
|
|
|
return SUCCESS; |
|
|
|
} |
|
|
|
|