From 210ee03cd54af6a18afa1428396911eb2e8590e8 Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Mon, 28 Dec 2020 10:58:06 +0800 Subject: [PATCH] display model info --- tests/ut/ge/session/omg_omg_unittest.cc | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/ut/ge/session/omg_omg_unittest.cc b/tests/ut/ge/session/omg_omg_unittest.cc index 542f4c61..918be70b 100644 --- a/tests/ut/ge/session/omg_omg_unittest.cc +++ b/tests/ut/ge/session/omg_omg_unittest.cc @@ -31,8 +31,23 @@ class UTEST_omg_omg : public testing::Test { void TearDown() override {} }; +TEST_F(UTEST_omg_omg, display_model_info_failed) +{ + ge::proto::ModelDef model_def; + PrintModelInfo(&model_def); +} + TEST_F(UTEST_omg_omg, display_model_info_success) { ge::proto::ModelDef model_def; + auto attrs = model_def.mutable_attr(); + ge::proto::AttrDef *attr_def_soc = &(*attrs)["soc_version"]; + attr_def_soc->set_s("Ascend310"); + ge::proto::AttrDef *attr_def = &(*attrs)["om_info_list"]; + attr_def->mutable_list()->add_i(1); + 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); }