From 8ae9370dbf8ecf17f4a528484b1b5509f179262f Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Wed, 23 Dec 2020 16:07:21 +0800 Subject: [PATCH] display model info --- ge/common/helper/model_helper.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc index aad348d6..e552424c 100644 --- a/ge/common/helper/model_helper.cc +++ b/ge/common/helper/model_helper.cc @@ -78,6 +78,8 @@ Status ModelHelper::SaveModelPartition(std::shared_ptr &om_fil Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer) { vector om_info; + ModelPtr model_tmp = ge::MakeShared(ge_model->GetName(), ge_model->GetPlatformVersion()); + (void)model_tmp->Save(model_buffer); GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize()); om_info.push_back(model_buffer.GetSize()); @@ -118,10 +120,10 @@ 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()); (void)model_tmp->Save(model_buffer); - SaveSizeToModelDef(ge_model, model_buffer); GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize()); if (model_buffer.GetSize() > 0) { if (SaveModelPartition(om_file_save_helper, ModelPartitionType::MODEL_DEF, model_buffer.GetData(),