Browse Source

add UT

pull/1313/head
chuxing 4 years ago
parent
commit
062590db32
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 5
- 0
tests/ut/ge/single_op/single_op_model_unittest.cc View File

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


Loading…
Cancel
Save