| @@ -24,7 +24,7 @@ const size_t kNumAddresses = 2; | |||||
| Status RtsKernelTaskBuilder::BuildMemcpyAsyncTask(const OpDescPtr &op_desc, | Status RtsKernelTaskBuilder::BuildMemcpyAsyncTask(const OpDescPtr &op_desc, | ||||
| const domi::MemcpyAsyncDef &kernel_def, | const domi::MemcpyAsyncDef &kernel_def, | ||||
| const SingleOpModelParam ¶m,r | |||||
| const SingleOpModelParam ¶m, | |||||
| std::unique_ptr<MemcpyAsyncTask> &task) { | std::unique_ptr<MemcpyAsyncTask> &task) { | ||||
| task.reset(new(std::nothrow)MemcpyAsyncTask()); | task.reset(new(std::nothrow)MemcpyAsyncTask()); | ||||
| GE_CHECK_NOTNULL(task); | 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/single_op_manager.cc" | ||||
| "${GE_CODE_DIR}/ge/single_op/task/aicpu_task_builder.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/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/tensor_value.cc" | ||||
| "${GE_CODE_DIR}/ge/hybrid/common/npu_memory_allocator.cc" | "${GE_CODE_DIR}/ge/hybrid/common/npu_memory_allocator.cc" | ||||
| "${GE_CODE_DIR}/ge/hybrid/executor/rt_callback_manager.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}); | op_desc->SetOutputOffset({0}); | ||||
| std::unique_ptr<MemcpyAsyncTask> task; | std::unique_ptr<MemcpyAsyncTask> task; | ||||
| SingleOpModelParam param{}; | SingleOpModelParam param{}; | ||||
| param.memory_size = 1024; | |||||
| auto ret = RtsKernelTaskBuilder::BuildMemcpyAsyncTask(op_desc, kernel_def, param, task); | auto ret = RtsKernelTaskBuilder::BuildMemcpyAsyncTask(op_desc, kernel_def, param, task); | ||||
| ASSERT_EQ(ret, SUCCESS); | ASSERT_EQ(ret, SUCCESS); | ||||