|
|
|
@@ -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); |
|
|
|
|