From c8e2ac3c1358b9b94ed8a6581e612109ef05f663 Mon Sep 17 00:00:00 2001 From: guopeian Date: Tue, 20 Jul 2021 15:55:25 +0800 Subject: [PATCH] fix --- tests/ut/ge/single_op/single_op_task_unittest.cc | 8 ++++++-- 1 file changed, 6 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 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); }