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 1d6b4390..9d7f9954 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -241,6 +241,7 @@ TEST_F(UtestSingleOpModel, test_build_memcpy_task) { param.memory_size = 1024; auto ret = RtsKernelTaskBuilder::BuildMemcpyAsyncTask(op_desc, kernel_def, param, task); ASSERT_EQ(ret, SUCCESS); + ASSERT_EQ(task->LaunchKernel(nullptr), SUCCESS); op_desc->SetInputOffset({0, 0}); op_desc->AddOutputDesc("y2", tensor_desc);