|
|
@@ -989,7 +989,7 @@ Status AiCpuCCTask::CopyDataToHbm(vector<DataBuffer> &outputs, |
|
|
|
|
|
|
|
auto ret = rtCpuKernelLaunchWithFlag(static_cast<const void *>(memcpy_so_name_.data()), |
|
|
|
static_cast<const void *>(memcpy_kernel_name_.data()), |
|
|
|
block_dim_, memcpy_args_.get(), static_cast<uint32_t>(memcpy_args_size_), |
|
|
|
block_dim_, memcpy_args_.get(), memcpy_args_size_, |
|
|
|
nullptr, stream, RT_KERNEL_DEFAULT); |
|
|
|
GE_CHK_RT_RET(ret); |
|
|
|
GE_CHK_RT_RET(rtStreamSynchronize(stream)); |
|
|
|