Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
a6140ebc17
1 changed files with 5 additions and 7 deletions
  1. +5
    -7
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 5
- 7
tests/ut/ge/single_op/single_op_model_unittest.cc View File

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

Loading…
Cancel
Save