From beb2a9ca51a6acf46451d0153543a3f2ebdd038f Mon Sep 17 00:00:00 2001
From: "wangwenhua1@huawei.com" <wangwenhua1@huawei.com>
Date: Wed, 23 Dec 2020 14:36:05 +0800
Subject: [PATCH] display model info

---
 ge/generator/ge_generator.cc | 4 ++--
 ge/session/omg.cc            | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index eb40bf5f..a3ad443d 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -556,7 +556,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
     return;
   }
 
-  std::cout << "------ Display Model Info start ------" << std::endl;
+  std::cout << "------------ Display Model Info start ------------" << std::endl;
   // system info
   std::string atc_version;
   (void)ge::AttrUtils::GetStr(obj, ATTR_MODEL_ATC_VERSION, atc_version);
@@ -610,7 +610,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
     std::cout << "Display Model Info error, please check!"  << std::endl;
   };
 
-  std::cout << "------ Display Model Info end ------"  << std::endl;
+  std::cout << "------------ Display Model Info end   ------------"  << std::endl;
 }
 
 Status GeGenerator::GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs,
diff --git a/ge/session/omg.cc b/ge/session/omg.cc
index 3a05467a..c9da9a40 100755
--- a/ge/session/omg.cc
+++ b/ge/session/omg.cc
@@ -909,7 +909,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
             << std::endl;
 
   // om info
-  iter = model_attr_map->find("om_info_size");
+  iter = model_attr_map->find("om_info_list");
   auto bt = iter->second.list().i_size();
   if (bt == 5) {
     std::cout << "om info: "