From 6031e5576c185d8658a11e085f9e3ddf7d9687ef Mon Sep 17 00:00:00 2001
From: "wangwenhua1@huawei.com" <wangwenhua1@huawei.com>
Date: Wed, 23 Dec 2020 15:22:11 +0800
Subject: [PATCH] display model info

---
 ge/session/omg.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ge/session/omg.cc b/ge/session/omg.cc
index c9da9a40..35939553 100755
--- a/ge/session/omg.cc
+++ b/ge/session/omg.cc
@@ -910,6 +910,12 @@ 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::endl;
+    std::cout << "------------ Display Model Info end   ------------"  << std::endl;
+    return;
+  }
   auto bt = iter->second.list().i_size();
   if (bt == 5) {
     std::cout << "om info: "