display model info

ge/common/helper/model_helper.cc

@@ -80,27 +80,18 @@ Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &m
   vector<int64_t> 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<ModelTaskDef> 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);