display model info

ge/generator/ge_generator.cc

@@ -49,6 +49,7 @@ const char *const kAIcoreEngine = "AIcoreEngine";
 const char *const kFileNameSuffix = "online";
 const size_t kDynamicDimSize = 1;
 const int64_t kDynamicDimValue = -2;
+const size_t kOmInfoSize = 5;
 
 std::map<ge::OpEngineType, std::string> engine_type_map{
     {ge::ENGINE_SYS, kEngineNameDefault}, {ge::ENGINE_AICORE, kAIcoreEngine}, {ge::ENGINE_VECTOR, kVectorEngine}};
@@ -602,7 +603,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
 
   vector<int64_t> om_info;
   (void)ge::AttrUtils::GetListInt(obj, "om_info_list", om_info);
-  if (om_info.size() == 5) {
+  if (om_info.size() == kOmInfoSize) {
     std::cout << "om info: "
               << "modeldef_size"
               << "[" << om_info[0] << "], "
@@ -615,7 +616,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
               << "task_info_size"
               << "[" << om_info[4] << "]." << std::endl;
   } else {
-    std::cout << "Display Model Info error, please check!"  << std::endl;
+    std::cout << "Display Model Info failed, please check!"  << std::endl;
   };
 
   std::cout << "------------ Display Model Info end   ------------"  << std::endl;

ge/session/omg.cc

@@ -71,6 +71,7 @@ const char *const kOutputTypeError = "The multiple out nodes set in output_type
 const size_t kNodeNameIndex = 0;
 const size_t kIndexStrIndex = 1;
 const size_t kDTValueIndex = 2;
+const size_t kOmInfoSize = 5;
 }  // namespace
 
 // When the model is converted to a JSON file, the following operator attributes in the blacklist will be ignored
@@ -916,8 +917,8 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
     std::cout << "------------ Display Model Info end   ------------"  << std::endl;
     return;
   }
-  auto bt = iter->second.list().i_size();
-  if (bt == 5) {
+  auto list_size = iter->second.list().i_size();
+  if (list_size == kOmInfoSize) {
     std::cout << "om info: "
               << "modeldef_size"
               << "[" << iter->second.list().i(0) << "], "