|
|
@@ -422,15 +422,13 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) { |
|
|
|
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(); |
|
|
|
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; |
|
|
|
ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); |
|
|
|
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); |
|
|
|
} |