Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
a134c4df32
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/task/op_task.cc

+ 1
- 1
ge/single_op/task/op_task.cc View File

@@ -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));


Loading…
Cancel
Save