diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 8fe3a3d5..e7f11893 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -511,13 +511,13 @@ bool GeGenerator::Impl::SetOmSystemInfo(AttrHolder &obj) { void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) { // - size_t memory_size; + int64_t memory_size; (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_MEMORY_SIZE, memory_size); - size_t weight_size; + int64_t weight_size; (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_WEIGHT_SIZE, weight_size); - size_t stream_num; + int64_t stream_num; (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_STREAM_NUM, stream_num); - size_t event_num; + int64_t event_num; (void)ge::AttrUtils::GetInt(obj, ATTR_MODEL_EVENT_NUM, event_num); std::cout << ATTR_MODEL_MEMORY_SIZE << "[" << memory_size << "], " << ATTR_MODEL_WEIGHT_SIZE << "[" << weight_size << "], " << ATTR_MODEL_STREAM_NUM << "[" << stream_num << "], " << ATTR_MODEL_EVENT_NUM << "[" << event_num << "]." << @@ -525,24 +525,24 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) { // std::string atc_version; - (void)ge::AttrUtils::GetStr(obj, ATTR_MODEL_ATC_VERSION, atc_version) + (void)ge::AttrUtils::GetStr(obj, ATTR_MODEL_ATC_VERSION, atc_version); std::string soc_version; - (void)ge::AttrUtils::GetStr(obj, "soc_version", soc_version) + (void)ge::AttrUtils::GetStr(obj, "soc_version", soc_version); std::string framework_type; - (void)ge::AttrUtils::GetStr(obj, "framework_type", framework_type) + (void)ge::AttrUtils::GetStr(obj, "framework_type", framework_type); std::cout << ATTR_MODEL_ATC_VERSION << "[" << atc_version << "], " << "soc_version" << "[" << soc_version << "], " << "framework_type" << "[" << framework_type << "]." << std::endl; // - size_t modeldef_size; + int64_t modeldef_size; (void)ge::AttrUtils::GetInt(obj, "modeldef_size", modeldef_size); - size_t weight_data_size; + int64_t weight_data_size; (void)ge::AttrUtils::GetInt(obj, "weight_data_size", weight_data_size); - size_t tbe_kernels_size; + int64_t tbe_kernels_size; (void)ge::AttrUtils::GetInt(obj, "tbe_kernels_size", tbe_kernels_size); - size_t cust_aicpu_kernel_store_size; + int64_t cust_aicpu_kernel_store_size; (void)ge::AttrUtils::GetInt(obj, "cust_aicpu_kernel_store_size", cust_aicpu_kernel_store_size); - size_t task_info_size; + int64_t task_info_size; (void)ge::AttrUtils::GetInt(obj, "task_info_size", task_info_size); std::cout << "modeldef_size" << "[" << modeldef_size << "], " << "weight_data_size" << "[" << weight_data_size << "], " << "tbe_kernels_size" << "[" << tbe_kernels_size << "], " << "cust_aicpu_kernel_store_size" << "[" << @@ -821,7 +821,7 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo if (!SetOppVersionInfo(*(model_root.get()))) { GELOGW("SetPackageVersionInfo of ops failed!"); } - if (!SetOmystemInfo(*(model_root.get()))) { + if (!SetOmSystemInfo(*(model_root.get()))) { GELOGW("SetOmsystemInfo failed!"); } ModelHelper model_helper;