|
|
@@ -645,6 +645,7 @@ Status KernelTaskInfo::InitTVMTask(uint16_t offset, const domi::KernelDef &kerne |
|
|
|
GE_CHECK_NOTNULL(op_desc); |
|
|
|
|
|
|
|
args_addr = std::unique_ptr<uint8_t[]>(new (std::nothrow) uint8_t[args_size_]); |
|
|
|
GE_CHECK_NOTNULL(args_addr); |
|
|
|
errno_t sec_ret = memcpy_s(args_addr.get(), args_size_, kernel_def.args().data(), args_size_); |
|
|
|
if (sec_ret != EOK) { |
|
|
|
REPORT_CALL_ERROR("E19999", "Call memcpy_s fail, size:%u, ret:0x%X", args_size_, sec_ret); |
|
|
@@ -1000,6 +1001,7 @@ Status KernelTaskInfo::InitAicpuTask(uint32_t op_index, const domi::KernelDef &k |
|
|
|
|
|
|
|
// copy args to new host memory |
|
|
|
args_addr = std::unique_ptr<uint8_t[]>(new (std::nothrow) uint8_t[args_size_]); |
|
|
|
GE_CHECK_NOTNULL(args_addr); |
|
|
|
GE_PRINT_DYNAMIC_MEMORY(new, "cce task physical memory.", sizeof(uint8_t) * args_size_) |
|
|
|
errno_t sec_ret = memcpy_s(args_addr.get(), args_size_, kernel_def.args().data(), args_size_); |
|
|
|
if (sec_ret != EOK) { |
|
|
|