|
|
@@ -499,8 +499,8 @@ Status AicpuNodeTask::CopyDataToHbm(TaskContext &context, |
|
|
|
|
|
|
|
|
RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] Start"); |
|
|
RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] Start"); |
|
|
uint32_t flag = RT_KERNEL_DEFAULT; |
|
|
uint32_t flag = RT_KERNEL_DEFAULT; |
|
|
auto rt_ret = rtCpuKernelLaunchWithFlag(reinterpret_cast<const void *>(memcpy_so_name.c_str()), |
|
|
|
|
|
reinterpret_cast<const void *>(memcpy_kernel_name.c_str()), |
|
|
|
|
|
|
|
|
auto rt_ret = rtCpuKernelLaunchWithFlag(reinterpret_cast<const void *>(memcpy_so_name_.c_str()), |
|
|
|
|
|
reinterpret_cast<const void *>(memcpy_kernel_name_.c_str()), |
|
|
1, // default core dim is 1 |
|
|
1, // default core dim is 1 |
|
|
memcpy_args_.get(), memcpy_args_size_, |
|
|
memcpy_args_.get(), memcpy_args_size_, |
|
|
nullptr, context.GetStream(), flag); |
|
|
nullptr, context.GetStream(), flag); |
|
|
@@ -721,7 +721,7 @@ Status AicpuNodeTask::SetMemCopyTask(const domi::TaskDef &task_def) { |
|
|
return FAILED;); |
|
|
return FAILED;); |
|
|
|
|
|
|
|
|
memcpy_args_.reset(new(std::nothrow) uint8_t[memcpy_args_size_]()); |
|
|
memcpy_args_.reset(new(std::nothrow) uint8_t[memcpy_args_size_]()); |
|
|
GE_IF_BOOL_EXEC(memcpy_task == nullptr, |
|
|
|
|
|
|
|
|
GE_IF_BOOL_EXEC(memcpy_args_ == nullptr, |
|
|
REPORT_INNER_ERROR("E19999", "new memory failed for Node[MemCopy], task_size[%u].", |
|
|
REPORT_INNER_ERROR("E19999", "new memory failed for Node[MemCopy], task_size[%u].", |
|
|
memcpy_args_size_); |
|
|
memcpy_args_size_); |
|
|
GELOGE(FAILED, "[Malloc][Memory] failed for Node[MemCopy], task_size[%u].", |
|
|
GELOGE(FAILED, "[Malloc][Memory] failed for Node[MemCopy], task_size[%u].", |
|
|
|