From 3c8509c54d388efa33809fc4bc39db68fbbdbc39 Mon Sep 17 00:00:00 2001 From: guopeian Date: Mon, 19 Jul 2021 15:12:16 +0800 Subject: [PATCH] fix --- ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc | 3 +-- ge/single_op/task/op_task.cc | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc index c15989fd..b73566c9 100755 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc @@ -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(memcpy_so_name_.c_str()), reinterpret_cast(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"); diff --git a/ge/single_op/task/op_task.cc b/ge/single_op/task/op_task.cc index e1d6e818..5bc33cfb 100755 --- a/ge/single_op/task/op_task.cc +++ b/ge/single_op/task/op_task.cc @@ -892,7 +892,7 @@ Status AiCpuCCTask::CopyDataToHbm(vector &outputs, auto ret = rtCpuKernelLaunchWithFlag(static_cast(memcpy_so_name_.data()), static_cast(memcpy_kernel_name_.data()), block_dim_, memcpy_args_.get(), static_cast(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;