Browse Source

display model info

pull/688/head
wangwenhua1@huawei.com 5 years ago
parent
commit
f5e40a9df5
2 changed files with 7 additions and 4 deletions
  1. +1
    -1
      ge/common/helper/model_helper.cc
  2. +6
    -3
      ge/generator/ge_generator.cc

+ 1
- 1
ge/common/helper/model_helper.cc View File

@@ -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);


+ 6
- 3
ge/generator/ge_generator.cc View File

@@ -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<int64_t> 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] << "], "


Loading…
Cancel
Save