|
|
|
@@ -871,7 +871,7 @@ void GetGroupName(ge::proto::ModelDef &model_def) { |
|
|
|
} |
|
|
|
|
|
|
|
FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { |
|
|
|
std::cout << "------------ Display Model Info start ------------" << std::endl; |
|
|
|
std::cout << "============ Display Model Info start ============" << std::endl; |
|
|
|
|
|
|
|
auto model_attr_map = model_def->mutable_attr(); |
|
|
|
// system info |
|
|
|
@@ -881,7 +881,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { |
|
|
|
auto soc_version = (iter != model_attr_map->end()) ? iter->second.s() : ""; |
|
|
|
iter = model_attr_map->find("framework_type"); |
|
|
|
auto framework_type = (iter != model_attr_map->end()) ? iter->second.s() : ""; |
|
|
|
std::cout << "system info: " |
|
|
|
std::cout << "system info: " |
|
|
|
<< ATTR_MODEL_ATC_VERSION |
|
|
|
<< "[" << atc_version << "], " |
|
|
|
<< "soc_version" |
|
|
|
@@ -919,7 +919,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { |
|
|
|
} |
|
|
|
auto list_size = iter->second.list().i_size(); |
|
|
|
if (list_size == kOmInfoSize) { |
|
|
|
std::cout << "om info: " |
|
|
|
std::cout << "om info: " |
|
|
|
<< "modeldef_size" |
|
|
|
<< "[" << iter->second.list().i(0) << " B], " |
|
|
|
<< "weight_data_size" |
|
|
|
|