zhangxiaokun
5 years ago
1 changed files with 3 additions and 3 deletions
Unified View
Diff Options
+ 3
- 3
tests/ut/ge/graph/load/davinci_model_unittest.cc
View File
| @@ -39,8 +39,8 @@ TEST_F(UtestDavinciModel, init_success) { | |||||
| GeModelPtr ge_model = make_shared<GeModel>(); | GeModelPtr ge_model = make_shared<GeModel>(); | ||||
| ge_model->SetGraph(GraphUtils::CreateGraphFromComputeGraph(graph)); | 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<domi::ModelTaskDef> model_task_def = make_shared<domi::ModelTaskDef>(); | shared_ptr<domi::ModelTaskDef> model_task_def = make_shared<domi::ModelTaskDef>(); | ||||
| ge_model->SetModelTaskDef(model_task_def); | ge_model->SetModelTaskDef(model_task_def); | ||||
| @@ -69,7 +69,7 @@ TEST_F(UtestDavinciModel, init_success) { | |||||
| op_output->SetSrcIndex( { 0 } ); | op_output->SetSrcIndex( { 0 } ); | ||||
| NodePtr node_output = graph->AddNode(op_output); | 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_stream_id(0); | ||||
| task_def->set_type(RT_MODEL_TASK_MEMCPY_ASYNC); | task_def->set_type(RT_MODEL_TASK_MEMCPY_ASYNC); | ||||
| domi::MemcpyAsyncDef *memcpy_async = task_def->mutable_memcpy_async(); | domi::MemcpyAsyncDef *memcpy_async = task_def->mutable_memcpy_async(); | ||||