|
|
|
@@ -241,6 +241,11 @@ 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); |
|
|
|
uintptr_t *addresses = nullptr; |
|
|
|
size_t arg_count = 0; |
|
|
|
task->GetIoAddr(addresses, arg_count); |
|
|
|
ASSERT_EQ(arg_count == 2); |
|
|
|
ASSERT_TRUE(addresses != nullptr); |
|
|
|
ASSERT_EQ(task->LaunchKernel(nullptr), SUCCESS); |
|
|
|
|
|
|
|
op_desc->SetInputOffset({0, 0}); |
|
|
|
|