From 8617dcc3043f9e3ecda06983ca5cb296ac5883c3 Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Thu, 18 Feb 2021 15:41:06 +0800 Subject: [PATCH] display model info --- tests/ut/ge/session/omg_omg_unittest.cc | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/ut/ge/session/omg_omg_unittest.cc b/tests/ut/ge/session/omg_omg_unittest.cc index b9c7f1ec..334df319 100644 --- a/tests/ut/ge/session/omg_omg_unittest.cc +++ b/tests/ut/ge/session/omg_omg_unittest.cc @@ -33,7 +33,7 @@ class UtestOmg : public testing::Test { TEST_F(UtestOmg, display_model_info_failed) { ge::proto::ModelDef model_def; - PrintModelInfo(&model_def); + PrintModelInfo(&model_def, 1); } TEST_F(UtestOmg, display_model_info_success) { @@ -46,7 +46,6 @@ TEST_F(UtestOmg, display_model_info_success) { attr_def->mutable_list()->add_i(2); attr_def->mutable_list()->add_i(3); attr_def->mutable_list()->add_i(4); - attr_def->mutable_list()->add_i(5); - PrintModelInfo(&model_def); + PrintModelInfo(&model_def, 1); } } // namespace ge