|
|
@@ -379,7 +379,10 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { |
|
|
|
ge_model->SetGraph(GraphUtils::CreateGraphFromComputeGraph(graph)); |
|
|
|
shared_ptr<domi::ModelTaskDef> model_task_def = make_shared<domi::ModelTaskDef>(); |
|
|
|
ge_model->SetModelTaskDef(model_task_def); |
|
|
|
|
|
|
|
|
|
|
|
AicpuTaskStruct args; |
|
|
|
args.head.length = sizeof(args); |
|
|
|
args.head.ioAddrNum = 6; |
|
|
|
domi::TaskDef *task_def = model_task_def->add_task(); |
|
|
|
task_def->set_type(RT_MODEL_TASK_KERNEL); |
|
|
|
domi::KernelDef *kernel_def = task_def->mutable_kernel(); |
|
|
|