add UT

4 years ago · 062590db32
parent 59c9181cdb
commit 062590db32
1 changed files with 5 additions and 0 deletions
--- 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,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});