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