Browse Source

fix ut

pull/2039/head
guopeian 3 years ago
parent
commit
2492b60d63
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 4
- 1
tests/ut/ge/single_op/single_op_model_unittest.cc View File

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


Loading…
Cancel
Save