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 57d2db13..356fc0e8 100644 --- a/tests/ut/ge/single_op/single_op_task_unittest.cc +++ b/tests/ut/ge/single_op/single_op_task_unittest.cc @@ -202,6 +202,7 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { auto op_desc = make_shared("deque", "Deque"); 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); task.SetOpDesc(op_desc); task.num_inputs_ = 2; task.num_outputs_ = 1; @@ -221,11 +222,13 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { std::string ext_info_str(ext_mem, sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); vector inputs(2, DataBuffer()); vector outputs(1, DataBuffer()); - vector inputs_desc(2, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT)); - vector outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT)); + vector inputs_desc(2, tensor); + vector outputs_desc(1, tensor); 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) {