diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc index a69bcc0b..aad348d6 100644 --- a/ge/common/helper/model_helper.cc +++ b/ge/common/helper/model_helper.cc @@ -101,7 +101,7 @@ Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &m size_t partition_task_size = model_task_def->ByteSizeLong(); GELOGD("SaveSizeToModelDef task_info_size is %zu", partition_task_size); om_info.push_back(partition_task_size); - + GE_CHK_BOOL_EXEC(ge::AttrUtils::SetListInt(ge_model, "om_info_list", om_info), GELOGE(FAILED, "SetListInt of om_info_list failed."); return FAILED); diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index efef82d4..eb40bf5f 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -564,7 +564,8 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) { (void)ge::AttrUtils::GetStr(obj, "soc_version", soc_version); std::string framework_type; (void)ge::AttrUtils::GetStr(obj, "framework_type", framework_type); - std::cout << ATTR_MODEL_ATC_VERSION + std::cout << "system info: " + << ATTR_MODEL_ATC_VERSION << "[" << atc_version << "], " << "soc_version" << "[" << soc_version << "], " @@ -580,7 +581,8 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) { (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_STREAM_NUM, stream_num); int64_t event_num; (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_EVENT_NUM, event_num); - std::cout << ATTR_MODEL_MEMORY_SIZE + std::cout << "resource info: " + << ATTR_MODEL_MEMORY_SIZE << "[" << memory_size << "], " << ATTR_MODEL_WEIGHT_SIZE << "[" << weight_size << "], " @@ -593,7 +595,8 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) { vector om_info; (void)ge::AttrUtils::GetListInt(obj, "om_info_list", om_info); if (om_info.size() == 5) { - std::cout << "modeldef_size" + std::cout << "om info: " + << "modeldef_size" << "[" << om_info[0] << "], " << "weight_data_size" << "[" << om_info[1] << "], "