From 42c51addd3a1a2010395fbc0978dbd80e29b2a4b Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Tue, 20 Jul 2021 16:57:24 +0800
Subject: [PATCH] fix

---
 tests/ut/ge/single_op/single_op_task_unittest.cc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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 356fc0e8..74dba70f 100644
--- a/tests/ut/ge/single_op/single_op_task_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_task_unittest.cc
@@ -203,6 +203,9 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
   ge::AttrUtils::SetBool(op_desc, ATTR_NAME_IS_BLOCKING_OP, false);
   AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
   GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
+  op_desc->AddInputDesc(tensor);
+  op_desc->AddInputDesc(tensor);
+  op_desc->AddOutputDesc(tensor);
   task.SetOpDesc(op_desc);
   task.num_inputs_ = 2;
   task.num_outputs_ = 1;
@@ -227,8 +230,6 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
   ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS);
   task.unknown_type_ = ge::DEPEND_COMPUTE;
   ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS);
-  task.num_inputs_ = 0;
-  task.num_outputs_ = 0;
   ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS);
 }
 TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) {