From 1868a9cc10b3c9b7b2b363fd3068d6e0bdb6d521 Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Tue, 22 Dec 2020 17:25:26 +0800 Subject: [PATCH] display model info --- ge/generator/ge_generator.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index d10d5673..99b41100 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -852,9 +852,6 @@ Status GeGenerator::Impl::SaveModel(const string &file_name_prefix, GeModelPtr & if (!SetOppVersionInfo(*(model.get()))) { GELOGW("SetPackageVersionInfo of ops failed!"); } - if (!SetOmSystemInfo(*(model.get()))) { - GELOGW("SetOmsystemInfo failed!"); - } ModelHelper model_helper; model_helper.SetSaveMode(is_offline_); Status ret = model_helper.SaveToOmModel(model, save_param_, file_name_prefix, model_buff); @@ -862,7 +859,6 @@ Status GeGenerator::Impl::SaveModel(const string &file_name_prefix, GeModelPtr & GELOGE(ret, "Save to om model failed"); return ret; } - DisplayModelInfo(*(model.get())); return SUCCESS; } @@ -894,6 +890,9 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo if (!SetOppVersionInfo(*(model_root.get()))) { GELOGW("SetPackageVersionInfo of ops failed!"); } + if (!SetOmSystemInfo(*(model_root.get()))) { + GELOGW("SetOmsystemInfo failed!"); + } ModelHelper model_helper; model_helper.SetSaveMode(is_offline_); ret = model_helper.SaveToOmRootModel(ge_root_model, save_param_, file_name_prefix, model_buff, is_unknown_shape); @@ -901,6 +900,7 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo GELOGE(ret, "Save to om model failed"); return ret; } + DisplayModelInfo(*(model_root.get())); return SUCCESS; }