@ -24,7 +24,7 @@ const size_t kNumAddresses = 2;
Status RtsKernelTaskBuilder::BuildMemcpyAsyncTask(const OpDescPtr &op_desc,
const domi::MemcpyAsyncDef &kernel_def,
const SingleOpModelParam ¶m,r
const SingleOpModelParam ¶m,
std::unique_ptr<MemcpyAsyncTask> &task) {
task.reset(new(std::nothrow)MemcpyAsyncTask());
GE_CHECK_NOTNULL(task);
@ -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"
@ -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);