From 501111b020bac4092468c38756b520ed160ece4e Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Thu, 7 Jan 2021 10:19:42 +0800 Subject: [PATCH] Eliminate variable_op_list_ --- tests/ut/ge/graph/load/davinci_model_unittest.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc index 0797ae24..38c5d837 100644 --- a/tests/ut/ge/graph/load/davinci_model_unittest.cc +++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc @@ -308,6 +308,9 @@ TEST_F(UtestDavinciModel, init_unknown) { TEST_F(UtestDavinciModel, Init_variable_op) { DavinciModel model(0, nullptr); + model.ge_model_ = make_shared(); + model.runtime_param_.mem_base = (uint8_t *)0x08000000; + model.runtime_param_.mem_size = 5120000; ComputeGraphPtr graph = make_shared("default"); GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); @@ -371,7 +374,7 @@ TEST_F(UtestDavinciModel, InitRealSizeAndShapeInfo_succ2) { GraphUtils::AddEdge(data1_node->GetOutDataAnchor(0), case1_node->GetInDataAnchor(0)); GraphUtils::AddEdge(case1_node->GetOutDataAnchor(0), output_node->GetInDataAnchor(0)); - + (void)AttrUtils::SetStr(output_node->GetOpDesc(), ATTR_ALL_GEARS_INFO, "1;2;4;8"); (void)AttrUtils::SetBool(case_node, ATTR_INSERT_BY_MBATCH, true);