Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
c300f3b2aa
2 changed files with 3 additions and 0 deletions
  1. +1
    -0
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
  2. +2
    -0
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 1
- 0
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -201,6 +201,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) {
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), SUCCESS);
node_item->num_outputs = 1;
AicpuNodeTask aicpu_node_task2(node_item, task_def);
ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
kernel_def->set_args_size(0);
ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
char* args2 = "123";


+ 2
- 0
tests/ut/ge/single_op/single_op_model_unittest.cc View File

@@ -424,6 +424,8 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) {
args.head.ioAddrNum = 6;
domi::KernelDef kernel_def;
kernel_def.set_args(reinterpret_cast<const char *>(&args), args.head.length);
kernel_def.set_args_size(args.head.length);
ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);
kernel_def.set_args_size(0);
AiCpuCCTask aicpu_task;
ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);


Loading…
Cancel
Save