From 59c9181cdbe47a07741a5582d5259882870aba01 Mon Sep 17 00:00:00 2001 From: chuxing Date: Mon, 22 Mar 2021 21:50:00 +0800 Subject: [PATCH] update UT --- tests/ut/ge/single_op/single_op_model_unittest.cc | 1 + 1 file changed, 1 insertion(+) 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);