From 3f4ba35e2022ccd506a5a272374d133f3cea5339 Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Mon, 19 Jul 2021 19:57:50 +0800
Subject: [PATCH] fix ut

---
 tests/ut/ge/single_op/single_op_model_unittest.cc |  7 -------
 tests/ut/ge/single_op/single_op_task_unittest.cc  | 22 ++++++++++++++++++++++
 2 files changed, 22 insertions(+), 7 deletions(-)

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 ee2d2c10..75db349e 100644
--- a/tests/ut/ge/single_op/single_op_model_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_model_unittest.cc
@@ -410,13 +410,6 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
   model.model_helper_.model_ = ge_model;
   auto op_desc = std::make_shared<ge::OpDesc>("add", "Add");
   AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
-  GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
-  for (int i = 0; i < 4; i++) {
-    op_desc->AddInputDesc(tensor);
-  }
-  for (int i = 0; i < 2; i++) {
-    op_desc->AddOutputDesc(tensor);
-  }
   NodePtr node = graph->AddNode(op_desc); 
   model.op_list_[0] = node;
   StreamResource *res = new (std::nothrow) StreamResource(1);
diff --git a/tests/ut/ge/single_op/single_op_task_unittest.cc b/tests/ut/ge/single_op/single_op_task_unittest.cc
index 8964df74..f88ef39a 100644
--- a/tests/ut/ge/single_op/single_op_task_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_task_unittest.cc
@@ -237,3 +237,25 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) {
     ASSERT_EQ(ret, PARAM_INVALID);
   }
 }
+TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
+  AiCpuCCTask task;
+  rtStream_t stream = nullptr;
+  task.num_inputs_ = 2;
+  task.num_outputs_ = 1;
+  task.input_is_const_ = {true, false};
+  int total_addr = 3;
+  uint32_t* addrs[total_addr] = {nullptr, nullptr, nullptr};
+  task.io_addr_ = reinterpret_cast<uintptr_t*>(addrs);
+  task.io_addr_num_ = total_addr;
+
+  {
+    vector<DataBuffer> inputs(2, DataBuffer());
+    vector<DataBuffer> outputs(1, DataBuffer());
+    vector<GeTensorDesc> inputs_desc(2, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT));
+    vector<GeTensorDesc> outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT));
+    task.unknown_type_ = ge::DEPEND_COMPUTE;
+    task.num_outputs_ = 1;
+    ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS);
+    ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS);
+  }
+}