Browse Source

Add UT

pull/803/head
zhangxiaokun 5 years ago
parent
commit
df2ea49ffb
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      tests/ut/ge/graph/load/davinci_model_unittest.cc

+ 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>();
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>();
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();


Loading…
Cancel
Save