From ed8774fb6d33b3bff48338f6cd0a04f396f5ad1c Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Wed, 23 Dec 2020 11:39:45 +0800 Subject: [PATCH] display model info --- ge/common/helper/model_helper.cc | 19 ++++--------------- 1 file changed, 4 insertions(+), 15 deletions(-) diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc index 9ad5a1bd..a69bcc0b 100644 --- a/ge/common/helper/model_helper.cc +++ b/ge/common/helper/model_helper.cc @@ -80,27 +80,18 @@ Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &m vector om_info; GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize()); om_info.push_back(model_buffer.GetSize()); -// GE_CHK_BOOL_EXEC(ge::AttrUtils::SetInt(ge_model, "modeldef_size", model_buffer.GetSize()), -// GELOGE(FAILED, "SetInt of modeldef_size failed."); -// return FAILED); + auto ge_model_weight = ge_model->GetWeight(); GELOGD("SaveSizeToModelDef weight_data_size is %zu, %p", ge_model_weight.GetSize(), ge_model_weight.GetData()); om_info.push_back(ge_model_weight.GetSize()); -// GE_CHK_BOOL_EXEC(ge::AttrUtils::SetInt(ge_model, "weight_data_size", ge_model_weight.GetSize()), -// GELOGE(FAILED, "SetInt of weight_data_size failed."); -// return FAILED); + TBEKernelStore tbe_kernel_store = ge_model->GetTBEKernelStore(); GELOGD("SaveSizeToModelDef tbe_kernels_size is %zu", tbe_kernel_store.DataSize()); om_info.push_back(tbe_kernel_store.DataSize()); -// GE_CHK_BOOL_EXEC(ge::AttrUtils::SetInt(ge_model, "tbe_kernels_size", tbe_kernel_store.DataSize()), -// GELOGE(FAILED, "SetInt of tbe_kernels_size failed."); -// return FAILED); + CustAICPUKernelStore cust_aicpu_kernel_store = ge_model->GetCustAICPUKernelStore(); GELOGD("SaveSizeToModelDef cust aicpu kernels size is %zu", cust_aicpu_kernel_store.DataSize()); om_info.push_back(cust_aicpu_kernel_store.DataSize()); -// GE_CHK_BOOL_EXEC(ge::AttrUtils::SetInt(ge_model, "cust_aicpu_kernel_store_size", cust_aicpu_kernel_store.DataSize()), -// GELOGE(FAILED, "SetInt of tbe_kernels_size failed."); -// return FAILED); std::shared_ptr model_task_def = ge_model->GetModelTaskDefPtr(); if (model_task_def == nullptr) { @@ -110,9 +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::SetInt(ge_model, "task_info_size", partition_task_size), -// GELOGE(FAILED, "SetInt of task_info_size failed."); -// return FAILED); + GE_CHK_BOOL_EXEC(ge::AttrUtils::SetListInt(ge_model, "om_info_list", om_info), GELOGE(FAILED, "SetListInt of om_info_list failed."); return FAILED);