diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc index 96424cb6..a3f03be3 100755 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc @@ -499,8 +499,8 @@ Status AicpuNodeTask::CopyDataToHbm(TaskContext &context, 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()), + 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); @@ -721,7 +721,7 @@ Status AicpuNodeTask::SetMemCopyTask(const domi::TaskDef &task_def) { return FAILED;); memcpy_args_.reset(new(std::nothrow) uint8_t[memcpy_args_size_]()); - GE_IF_BOOL_EXEC(memcpy_task == nullptr, + GE_IF_BOOL_EXEC(memcpy_args_ == nullptr, REPORT_INNER_ERROR("E19999", "new memory failed for Node[MemCopy], task_size[%u].", memcpy_args_size_); GELOGE(FAILED, "[Malloc][Memory] failed for Node[MemCopy], task_size[%u].", diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h index 2bfdd615..90e7791b 100644 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h @@ -67,8 +67,6 @@ class AicpuNodeTaskBase : public NodeTask { virtual Status CopyDataToHbm(TaskContext &context, const std::vector> &out_shape_hbm) = 0; -private: - Status TaskCallback(TaskContext &context); /// /// read result summary and prepare copy task memory.