From c300f3b2aac083885461f5f960fd8a276e2af419 Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Wed, 21 Jul 2021 15:56:49 +0800
Subject: [PATCH] fix ut

---
 tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc | 1 +
 tests/ut/ge/single_op/single_op_model_unittest.cc                      | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
index 61427594..402170c3 100644
--- a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
+++ b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
@@ -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";
diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc
index a6466972..666c31c8 100644
--- a/tests/ut/ge/single_op/single_op_model_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_model_unittest.cc
@@ -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);