From 062590db327ce58f714a23ed509ce645f385bfb4 Mon Sep 17 00:00:00 2001
From: chuxing <chuxing@huawei.com>
Date: Mon, 22 Mar 2021 22:00:09 +0800
Subject: [PATCH] add UT

---
 tests/ut/ge/single_op/single_op_model_unittest.cc | 5 +++++
 1 file changed, 5 insertions(+)

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 9d7f9954..fdf51d20 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,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});