|
|
@@ -915,8 +915,8 @@ Status KernelTaskInfo::InitAicpuTask(uint32_t op_index, const domi::KernelDef &k |
|
|
|
GELOGI("Node[%s] type[%s] kernel_ext_info size=%zu, aicpu_ext_info_addr_=%p", op_desc_->GetName().c_str(), |
|
|
|
op_desc_->GetType().c_str(), ext_info.size(), aicpu_ext_info_addr_); |
|
|
|
|
|
|
|
aicpu_param_head->extInfoAddr = static_cast<uintptr_t>(aicpu_ext_info_addr_); |
|
|
|
aicpu_param_head->extInfoLength = reinterpret_cast<uintptr_t>(ext_info.size()); |
|
|
|
aicpu_param_head->extInfoAddr = reinterpret_cast<uintptr_t>(aicpu_ext_info_addr_); |
|
|
|
aicpu_param_head->extInfoLength = static_cast<uintptr_t>(ext_info.size()); |
|
|
|
|
|
|
|
// malloc device memory for args |
|
|
|
rtError_t rt_ret = rtMalloc(static_cast<void **>(&args_), args_size_, RT_MEMORY_HBM); |
|
|
|