Browse Source

fix

pull/1879/head
guopeian 4 years ago
parent
commit
ce803f8a80
2 changed files with 3 additions and 5 deletions
  1. +3
    -3
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc
  2. +0
    -2
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.h

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

@@ -499,8 +499,8 @@ Status AicpuNodeTask::CopyDataToHbm(TaskContext &context,


RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] Start"); RECORD_CALLBACK_EVENT(context.GetExecutionContext(), node_name_.c_str(), "[LaunchCopy] Start");
uint32_t flag = RT_KERNEL_DEFAULT; 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()),
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 1, // default core dim is 1
memcpy_args_.get(), memcpy_args_size_, memcpy_args_.get(), memcpy_args_size_,
nullptr, context.GetStream(), flag); nullptr, context.GetStream(), flag);
@@ -721,7 +721,7 @@ Status AicpuNodeTask::SetMemCopyTask(const domi::TaskDef &task_def) {
return FAILED;); return FAILED;);


memcpy_args_.reset(new(std::nothrow) uint8_t[memcpy_args_size_]()); 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].", REPORT_INNER_ERROR("E19999", "new memory failed for Node[MemCopy], task_size[%u].",
memcpy_args_size_); memcpy_args_size_);
GELOGE(FAILED, "[Malloc][Memory] failed for Node[MemCopy], task_size[%u].", GELOGE(FAILED, "[Malloc][Memory] failed for Node[MemCopy], task_size[%u].",


+ 0
- 2
ge/hybrid/node_executor/aicpu/aicpu_node_executor.h View File

@@ -67,8 +67,6 @@ class AicpuNodeTaskBase : public NodeTask {


virtual Status CopyDataToHbm(TaskContext &context, virtual Status CopyDataToHbm(TaskContext &context,
const std::vector<std::unique_ptr<TensorBuffer>> &out_shape_hbm) = 0; const std::vector<std::unique_ptr<TensorBuffer>> &out_shape_hbm) = 0;
private:
Status TaskCallback(TaskContext &context);


/// ///
/// read result summary and prepare copy task memory. /// read result summary and prepare copy task memory.


Loading…
Cancel
Save