From f6dfe66f3b8f45eeb89f158ac86fe0ec5970ca62 Mon Sep 17 00:00:00 2001
From: chuxing <chuxing@huawei.com>
Date: Sat, 20 Mar 2021 18:55:02 +0800
Subject: [PATCH] add UT

---
 ge/single_op/task/rts_kernel_task_builder.cc      | 2 +-
 tests/ut/ge/CMakeLists.txt                        | 1 +
 tests/ut/ge/single_op/single_op_model_unittest.cc | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/ge/single_op/task/rts_kernel_task_builder.cc b/ge/single_op/task/rts_kernel_task_builder.cc
index 9d476e60..f48070a6 100644
--- a/ge/single_op/task/rts_kernel_task_builder.cc
+++ b/ge/single_op/task/rts_kernel_task_builder.cc
@@ -24,7 +24,7 @@ const size_t kNumAddresses = 2;
 
 Status RtsKernelTaskBuilder::BuildMemcpyAsyncTask(const OpDescPtr &op_desc,
                                                   const domi::MemcpyAsyncDef &kernel_def,
-                                                  const SingleOpModelParam &param,r
+                                                  const SingleOpModelParam &param,
                                                   std::unique_ptr<MemcpyAsyncTask> &task) {
   task.reset(new(std::nothrow)MemcpyAsyncTask());
   GE_CHECK_NOTNULL(task);
diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt
index 3caba788..b7a0066e 100755
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@@ -581,6 +581,7 @@ set(SINGLE_OP_SRC_FILES
     "${GE_CODE_DIR}/ge/single_op/single_op_manager.cc"
     "${GE_CODE_DIR}/ge/single_op/task/aicpu_task_builder.cc"
     "${GE_CODE_DIR}/ge/single_op/task/aicpu_kernel_task_builder.cc"
+    "${GE_CODE_DIR}/ge/single_op/task/rts_kernel_task_builder.cc"
     "${GE_CODE_DIR}/ge/hybrid/common/tensor_value.cc"
     "${GE_CODE_DIR}/ge/hybrid/common/npu_memory_allocator.cc"
     "${GE_CODE_DIR}/ge/hybrid/executor/rt_callback_manager.cc"
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 59f00fc8..1d6b4390 100644
--- a/tests/ut/ge/single_op/single_op_model_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_model_unittest.cc
@@ -238,6 +238,7 @@ TEST_F(UtestSingleOpModel, test_build_memcpy_task) {
   op_desc->SetOutputOffset({0});
   std::unique_ptr<MemcpyAsyncTask> task;
   SingleOpModelParam param{};
+  param.memory_size = 1024;
   auto ret = RtsKernelTaskBuilder::BuildMemcpyAsyncTask(op_desc, kernel_def, param, task);
   ASSERT_EQ(ret, SUCCESS);