| @@ -422,15 +422,13 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) { | |||||
| AicpuTaskStruct args; | AicpuTaskStruct args; | ||||
| args.head.length = sizeof(args); | args.head.length = sizeof(args); | ||||
| args.head.ioAddrNum = 6; | 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(); | |||||
| kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length); | |||||
| kernel_def->set_args_size(0); | |||||
| domi::KernelDef kernel_def; | |||||
| kernel_def.set_args(reinterpret_cast<const char *>(&args), args.head.length); | |||||
| kernel_def.set_args_size(0); | |||||
| AiCpuCCTask aicpu_task; | AiCpuCCTask aicpu_task; | ||||
| ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); | ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); | ||||
| char* args2 = "123"; | char* args2 = "123"; | ||||
| kernel_def->set_args(reinterpret_cast<const char *>(&args2), 3); | |||||
| kernel_def->set_args_size(3); | |||||
| kernel_def.set_args(reinterpret_cast<const char *>(&args2), 3); | |||||
| kernel_def.set_args_size(3); | |||||
| ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); | ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); | ||||
| } | } | ||||