fix ut

tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc

@@ -177,6 +177,31 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
 
 }
 
+TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) {
+  std::unique_ptr<NodeItem> new_node;
+  ASSERT_EQ(NodeItem::Create(node, new_node), SUCCESS);
+  NodeItem *node_item = new_node.get();
+  AicpuTaskStruct args;
+  args.head.length = sizeof(args);
+  args.head.ioAddrNum = 6;
+
+  domi::TaskDef task_def;
+  task_def.set_type(RT_MODEL_TASK_ALL_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(args.head.length);
+  node_item->num_outputs = 0;
+  AicpuNodeTask aicpu_node_task(node_item, task_def);
+  ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), SUCCESS);
+  aicpu_node_task->node_item_->num_outputs = 1;
+  kernel_def->set_args_size(0);
+  ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED);
+  char* args2 = "123";
+  kernel_def->set_args(reinterpret_cast<const char *>(&args2), 3);
+  kernel_def->set_args_size(3);
+  ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED);
+}
+
 TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {
   ComputeGraphPtr graph = std::make_shared<ComputeGraph>("test");
   GeRootModelPtr ge_root_model = std::make_shared<GeRootModel>(graph);

tests/ut/ge/single_op/single_op_model_unittest.cc

@@ -417,3 +417,20 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
   op_desc->SetOpKernelLibName(kEngineNameAiCpu);
   model.BuildTaskListForDynamicOp(res, single_op);
 }
+
+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);
+  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);
+  ASSERT_EQ(aicpu_task。SetMemCopyTask(kernel_def), FAILED);
+}