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