From df2ea49ffba8631c6145aa4df829fd53797505b9 Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Tue, 29 Dec 2020 22:10:28 +0800 Subject: [PATCH] Add UT --- tests/ut/ge/graph/load/davinci_model_unittest.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc index 37516c6e..ae33adf1 100644 --- a/tests/ut/ge/graph/load/davinci_model_unittest.cc +++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc @@ -39,8 +39,8 @@ TEST_F(UtestDavinciModel, init_success) { GeModelPtr ge_model = make_shared(); ge_model->SetGraph(GraphUtils::CreateGraphFromComputeGraph(graph)); - AttrUtils::SetInt(ge_model, ATTR_NODEL_MEMORY_SIZE, 5120000); - AttrUtils::SetInt(ge_model, ATTR_NODEL_STREAM_NUM, 1); + AttrUtils::SetInt(ge_model, ATTR_MODEL_MEMORY_SIZE, 5120000); + AttrUtils::SetInt(ge_model, ATTR_MODEL_STREAM_NUM, 1); shared_ptr model_task_def = make_shared(); ge_model->SetModelTaskDef(model_task_def); @@ -69,7 +69,7 @@ TEST_F(UtestDavinciModel, init_success) { op_output->SetSrcIndex( { 0 } ); NodePtr node_output = graph->AddNode(op_output); - domi::TaskDef *task_def = model_task_def->AddTask(); + domi::TaskDef *task_def = model_task_def->add_task(); task_def->set_stream_id(0); task_def->set_type(RT_MODEL_TASK_MEMCPY_ASYNC); domi::MemcpyAsyncDef *memcpy_async = task_def->mutable_memcpy_async();