diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 81d4a8cb..ff55e968 100644 --- a/ge/generator/ge_generator.cc +++ b/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 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 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; diff --git a/ge/session/omg.cc b/ge/session/omg.cc index c376f682..2a2f3274 100755 --- a/ge/session/omg.cc +++ b/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) << "], "