| @@ -308,6 +308,9 @@ TEST_F(UtestDavinciModel, init_unknown) { | |||||
| TEST_F(UtestDavinciModel, Init_variable_op) { | TEST_F(UtestDavinciModel, Init_variable_op) { | ||||
| DavinciModel model(0, nullptr); | DavinciModel model(0, nullptr); | ||||
| model.ge_model_ = make_shared<GeModel>(); | |||||
| model.runtime_param_.mem_base = (uint8_t *)0x08000000; | |||||
| model.runtime_param_.mem_size = 5120000; | |||||
| ComputeGraphPtr graph = make_shared<ComputeGraph>("default"); | ComputeGraphPtr graph = make_shared<ComputeGraph>("default"); | ||||
| GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); | 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(data1_node->GetOutDataAnchor(0), case1_node->GetInDataAnchor(0)); | ||||
| GraphUtils::AddEdge(case1_node->GetOutDataAnchor(0), output_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::SetStr(output_node->GetOpDesc(), ATTR_ALL_GEARS_INFO, "1;2;4;8"); | ||||
| (void)AttrUtils::SetBool(case_node, ATTR_INSERT_BY_MBATCH, true); | (void)AttrUtils::SetBool(case_node, ATTR_INSERT_BY_MBATCH, true); | ||||