display model info

ge/common/helper/model_helper.cc

@@ -76,9 +76,12 @@ Status ModelHelper::SaveModelPartition(std::shared_ptr<OmFileSaveHelper> &om_fil
   return SUCCESS;
 }
 
-Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer) {
+Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model) {
   vector<int64_t> om_info;
   ModelPtr model_tmp = ge::MakeShared<ge::Model>(ge_model->GetName(), ge_model->GetPlatformVersion());
+  model_tmp->SetVersion(ge_model->GetVersion());
+  model_tmp->SetAttr(ge_model->MutableAttrMap());
+  ge::Buffer model_buffer
   (void)model_tmp->Save(model_buffer);
   GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize());
   om_info.push_back(model_buffer.GetSize());
@@ -120,8 +123,8 @@ Status ModelHelper::SaveModelDef(std::shared_ptr<OmFileSaveHelper> &om_file_save
   }
   model_tmp->SetGraph(ge_model->GetGraph());
   model_tmp->SetVersion(ge_model->GetVersion());
-  SaveSizeToModelDef(ge_model, model_buffer);
   model_tmp->SetAttr(ge_model->MutableAttrMap());
+  SaveSizeToModelDef(ge_model);
 
   (void)model_tmp->Save(model_buffer);
   GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize());

ge/session/omg.cc

@@ -911,7 +911,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
   // om info
   iter = model_attr_map->find("om_info_list");
   if (iter == model_attr_map->end()) {
-    std::cout << "Display Model Info error, om_info_list is not find in om, please check atc version is matched."
+    std::cout << "Display Model Info failed, attr \"om_info_list\" is not found in om, check the version is matched."
               << std::endl;
     std::cout << "------------ Display Model Info end   ------------"  << std::endl;
     return;

inc/framework/common/helper/model_helper.h

@@ -84,7 +84,7 @@ class ModelHelper {
                             const uint8_t *data, size_t size, size_t model_index);
   Status SaveModelDef(shared_ptr<OmFileSaveHelper> &om_file_save_helper, const GeModelPtr &ge_model,
                       Buffer &model_buffer, size_t model_index = 0);
-  Status SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer);
+  Status SaveSizeToModelDef(const GeModelPtr &ge_model);
   Status SaveModelWeights(shared_ptr<OmFileSaveHelper> &om_file_save_helper, const GeModelPtr &ge_model,
                           size_t model_index = 0);
   Status SaveModelTbeKernel(shared_ptr<OmFileSaveHelper> &om_file_save_helper, const GeModelPtr &ge_model,