|
|
@@ -477,12 +477,11 @@ Status AicpuNodeTask::CopyDataToHbm(TaskContext &context, |
|
|
|
GE_CHK_STATUS_RET_NOLOG(PrepareCopyInputs(context, out_shape_hbm)); |
|
|
|
|
|
|
|
RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] Start"); |
|
|
|
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()), |
|
|
|
1, // default core dim is 1 |
|
|
|
memcpy_args_.get(), memcpy_args_size_, |
|
|
|
nullptr, context.GetStream(), flag); |
|
|
|
nullptr, context.GetStream(), RT_KERNEL_DEFAULT); |
|
|
|
GE_CHK_RT_RET(rt_ret); |
|
|
|
|
|
|
|
RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] End"); |
|
|
|