| @@ -703,50 +703,55 @@ Status AicpuNodeTask::SetMemCopyTask(const domi::TaskDef &task_def) { | |||||
| memcpy_args_size_ = kernel_def.args_size(); | memcpy_args_size_ = kernel_def.args_size(); | ||||
| memcpy_so_name_ = kernel_def.so_name(); | memcpy_so_name_ = kernel_def.so_name(); | ||||
| memcpy_kernel_name_ = kernel_def.kernel_name(); | memcpy_kernel_name_ = kernel_def.kernel_name(); | ||||
| GE_IF_BOOL_EXEC(memcpy_args.size() != memcpy_args_size_, | |||||
| REPORT_INNER_ERROR("E19999", "MemCopy task def args.size=%zu, but args_size=%u not equal.", | |||||
| memcpy_args.size(), memcpy_args_size_); | |||||
| GELOGE(FAILED, "[Check][Size]MemCopy task def args.size=%zu, but args_size=%u not equal.", | |||||
| memcpy_args.size(), memcpy_args_size_); | |||||
| return FAILED;); | |||||
| GE_IF_BOOL_EXEC(memcpy_args_size_ < sizeof(aicpu::AicpuParamHead), | |||||
| REPORT_INNER_ERROR("E19999", | |||||
| "Task def args_size=%u is less than aicpu param head len=%zu.", | |||||
| memcpy_args_size_, sizeof(aicpu::AicpuParamHead)); | |||||
| GELOGE(FAILED, | |||||
| "[Check][Size] Task def args_size=%u is less than aicpu param head len=%zu.", | |||||
| memcpy_args_size_, sizeof(aicpu::AicpuParamHead)); | |||||
| return FAILED;); | |||||
| if (memcpy_args.size() != memcpy_args_size_) { | |||||
| REPORT_INNER_ERROR("E19999", "MemCopy task def args.size=%zu, but args_size=%u not equal.", | |||||
| memcpy_args.size(), memcpy_args_size_); | |||||
| GELOGE(FAILED, "[Check][Size]MemCopy task def args.size=%zu, but args_size=%u not equal.", | |||||
| memcpy_args.size(), memcpy_args_size_); | |||||
| return FAILED; | |||||
| } | |||||
| if (memcpy_args_size_ < sizeof(aicpu::AicpuParamHead) { | |||||
| REPORT_INNER_ERROR("E19999", "Task def args_size=%u is less than aicpu param head len=%zu.", | |||||
| memcpy_args_size_, sizeof(aicpu::AicpuParamHead)); | |||||
| GELOGE(FAILED, "[Check][Size] Task def args_size=%u is less than aicpu param head len=%zu.", | |||||
| memcpy_args_size_, sizeof(aicpu::AicpuParamHead)); | |||||
| 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_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].", | |||||
| memcpy_args_size_); | |||||
| return FAILED;); | |||||
| if (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].", | |||||
| memcpy_args_size_); | |||||
| return FAILED; | |||||
| } | |||||
| errno_t sec_ret = memcpy_s(memcpy_args_.get(), memcpy_args_size_, memcpy_args.c_str(), memcpy_args.size()); | errno_t sec_ret = memcpy_s(memcpy_args_.get(), memcpy_args_size_, memcpy_args.c_str(), memcpy_args.size()); | ||||
| GE_IF_BOOL_EXEC(sec_ret != EOK, | |||||
| REPORT_INNER_ERROR("E19999", | |||||
| "memcpy_s argc_ failed for Node[MemCopy], ret: %d", sec_ret); | |||||
| GELOGE(INTERNAL_ERROR, | |||||
| "[Update][args] failed for Node[MemCopy], ret: %d", sec_ret); | |||||
| return sec_ret;); | |||||
| if (sec_ret != EOK) { | |||||
| REPORT_INNER_ERROR("E19999", | |||||
| "memcpy_s argc_ failed for Node[MemCopy], ret: %d", sec_ret); | |||||
| GELOGE(INTERNAL_ERROR, | |||||
| "[Update][args] failed for Node[MemCopy], ret: %d", sec_ret); | |||||
| return sec_ret; | |||||
| } | |||||
| auto memcpy_param_head = reinterpret_cast<aicpu::AicpuParamHead *>(memcpy_args_.get()); | auto memcpy_param_head = reinterpret_cast<aicpu::AicpuParamHead *>(memcpy_args_.get()); | ||||
| uint32_t memcpy_io_num = memcpy_param_head->ioAddrNum; | uint32_t memcpy_io_num = memcpy_param_head->ioAddrNum; | ||||
| auto memcpy_io_addr = memcpy_args_.get() + sizeof(aicpu::AicpuParamHead); | auto memcpy_io_addr = memcpy_args_.get() + sizeof(aicpu::AicpuParamHead); | ||||
| // if has input and output, need copy to ioaddr | // if has input and output, need copy to ioaddr | ||||
| int cpy_ret = memcpy_s(memcpy_io_addr, memcpy_args_size_ - sizeof(aicpu::AicpuParamHead), | int cpy_ret = memcpy_s(memcpy_io_addr, memcpy_args_size_ - sizeof(aicpu::AicpuParamHead), | ||||
| ©_io_addr_[0], sizeof(uint64_t) * memcpy_io_num); | ©_io_addr_[0], sizeof(uint64_t) * memcpy_io_num); | ||||
| GE_IF_BOOL_EXEC(cpy_ret != 0, | |||||
| REPORT_INNER_ERROR("E19999", "Node[Memcpoy] memcpy io addr to AicpuParamHead failed," | |||||
| "ret=%d, args_size=%u, io nums=%u.", | |||||
| cpy_ret, memcpy_args_size_, memcpy_io_num); | |||||
| GELOGE(INTERNAL_ERROR, "[Update][io_addr]Node[MemCopy] memcpy io addr to AicpuParamHead failed," | |||||
| "ret=%d, args_size=%u, io nums=%u.", | |||||
| cpy_ret, memcpy_args_size_, memcpy_io_num); | |||||
| return INTERNAL_ERROR;); | |||||
| if (cpy_ret != 0) { | |||||
| REPORT_INNER_ERROR("E19999", "Node[Memcpoy] memcpy io addr to AicpuParamHead failed," | |||||
| "ret=%d, args_size=%u, io nums=%u.", | |||||
| cpy_ret, memcpy_args_size_, memcpy_io_num); | |||||
| GELOGE(INTERNAL_ERROR, "[Update][io_addr]Node[MemCopy] memcpy io addr to AicpuParamHead failed," | |||||
| "ret=%d, args_size=%u, io nums=%u.", | |||||
| cpy_ret, memcpy_args_size_, memcpy_io_num); | |||||
| return INTERNAL_ERROR; | |||||
| } | |||||
| GELOGD("Set memcpy task for node[MemCopy] successfully."); | GELOGD("Set memcpy task for node[MemCopy] successfully."); | ||||
| return SUCCESS; | return SUCCESS; | ||||
| } | } | ||||