From fa59709e2a1865d29207ba306f93598e3ff20b88 Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Sat, 17 Jul 2021 16:36:48 +0800
Subject: [PATCH] fix

---
 .../ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc   | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

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 b225949b..1b15914b 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
@@ -151,18 +151,16 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
   task_def2.mutable_kernel()->set_args(reinterpret_cast<const char *>(&args), args.head.length);
   task_def2.mutable_kernel()->set_args_size(args.head.length);
 
-  hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2});
+  hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2, task_def2});
 
   AicpuNodeTask aicpu_node_task(node_item, task_def);
-  ASSERT_EQ(aicpu_node_task.Init(hybrid_model), FAILED);
+  ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS);
   ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS);
 
-
   //kernel_ex_def->set_allocated_kernel_ext_info(nullptr);
 
   free(ext_mem);
 
 }
-
 }  // namespace ge