Browse Source

fix

pull/2039/head
guopeian 3 years ago
parent
commit
3c8509c54d
2 changed files with 2 additions and 3 deletions
  1. +1
    -2
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc
  2. +1
    -1
      ge/single_op/task/op_task.cc

+ 1
- 2
ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc View File

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


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

@@ -892,7 +892,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_),
nullptr, stream, dump_flag_);
nullptr, stream, RT_KERNEL_DEFAULT);
GE_CHK_RT_RET(ret);
GE_CHK_RT_RET(rtStreamSynchronize(stream));
return SUCCESS;


Loading…
Cancel
Save