From 441493ad0ea63bf375ca9670f53bbce1f877817d Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Thu, 24 Dec 2020 09:24:16 +0800 Subject: [PATCH] display model info --- ge/common/helper/model_helper.cc | 7 +++++-- ge/session/omg.cc | 2 +- inc/framework/common/helper/model_helper.h | 2 +- 3 files changed, 7 insertions(+), 4 deletions(-) diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc index 454e6464..7e54a224 100644 --- a/ge/common/helper/model_helper.cc +++ b/ge/common/helper/model_helper.cc @@ -76,9 +76,12 @@ Status ModelHelper::SaveModelPartition(std::shared_ptr &om_fil return SUCCESS; } -Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer) { +Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model) { vector om_info; ModelPtr model_tmp = ge::MakeShared(ge_model->GetName(), ge_model->GetPlatformVersion()); + model_tmp->SetVersion(ge_model->GetVersion()); + model_tmp->SetAttr(ge_model->MutableAttrMap()); + ge::Buffer model_buffer (void)model_tmp->Save(model_buffer); GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize()); om_info.push_back(model_buffer.GetSize()); @@ -120,8 +123,8 @@ Status ModelHelper::SaveModelDef(std::shared_ptr &om_file_save } model_tmp->SetGraph(ge_model->GetGraph()); model_tmp->SetVersion(ge_model->GetVersion()); - SaveSizeToModelDef(ge_model, model_buffer); model_tmp->SetAttr(ge_model->MutableAttrMap()); + SaveSizeToModelDef(ge_model); (void)model_tmp->Save(model_buffer); GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize()); diff --git a/ge/session/omg.cc b/ge/session/omg.cc index 35939553..c376f682 100755 --- a/ge/session/omg.cc +++ b/ge/session/omg.cc @@ -911,7 +911,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { // om info iter = model_attr_map->find("om_info_list"); if (iter == model_attr_map->end()) { - std::cout << "Display Model Info error, om_info_list is not find in om, please check atc version is matched." + std::cout << "Display Model Info failed, attr \"om_info_list\" is not found in om, check the version is matched." << std::endl; std::cout << "------------ Display Model Info end ------------" << std::endl; return; diff --git a/inc/framework/common/helper/model_helper.h b/inc/framework/common/helper/model_helper.h index 5493b7b9..c609347b 100644 --- a/inc/framework/common/helper/model_helper.h +++ b/inc/framework/common/helper/model_helper.h @@ -84,7 +84,7 @@ class ModelHelper { const uint8_t *data, size_t size, size_t model_index); Status SaveModelDef(shared_ptr &om_file_save_helper, const GeModelPtr &ge_model, Buffer &model_buffer, size_t model_index = 0); - Status SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer); + Status SaveSizeToModelDef(const GeModelPtr &ge_model); Status SaveModelWeights(shared_ptr &om_file_save_helper, const GeModelPtr &ge_model, size_t model_index = 0); Status SaveModelTbeKernel(shared_ptr &om_file_save_helper, const GeModelPtr &ge_model,