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 e4f72f9b..57d2db13 100644 --- a/tests/ut/ge/single_op/single_op_task_unittest.cc +++ b/tests/ut/ge/single_op/single_op_task_unittest.cc @@ -197,7 +197,12 @@ TEST_F(UtestSingleOpTask, test_atomic_exec) { } TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { AiCpuCCTask task; - rtStream_t stream = nullptr; + rtStream_t stream; + ASSERT_EQ(rtStreamCreate(&stream, 0), RT_ERROR_NONE); + 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); + task.SetOpDesc(op_desc); task.num_inputs_ = 2; task.num_outputs_ = 1; task.input_is_const_ = {true, false}; @@ -220,7 +225,6 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { vector outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT)); ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS); 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); }