From d248531bc111ea3d1beb5e5e928fe22d665a8a5d Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Thu, 24 Dec 2020 11:40:28 +0800 Subject: [PATCH] display model info --- ge/generator/ge_generator.cc | 2 +- ge/offline/main.cc | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index ff55e968..a509c879 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -928,7 +928,7 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo GELOGE(ret, "Save to om model failed"); return ret; } - DisplayModelInfo(*(model_root.get())); + // DisplayModelInfo(*(model_root.get())); return SUCCESS; } diff --git a/ge/offline/main.cc b/ge/offline/main.cc index 74837e7a..55478991 100755 --- a/ge/offline/main.cc +++ b/ge/offline/main.cc @@ -1180,6 +1180,11 @@ domi::Status GenerateOmModel() { return domi::FAILED; } + if (FLAGS_display_model_info == "1") { + GELOGI("need to display model info."); + return ge::ConvertOm(FLAGS_output.c_str(), "", false); + } + return domi::SUCCESS; }