display model info

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