| @@ -3098,15 +3098,14 @@ Status DavinciModel::MallocKnownArgs() { | |||||
| GE_CHK_STATUS_RET_NOLOG(CheckCapability(FEATURE_TYPE_MEMORY, MEMORY_INFO_TS_4G_LIMITED, is_support)); | GE_CHK_STATUS_RET_NOLOG(CheckCapability(FEATURE_TYPE_MEMORY, MEMORY_INFO_TS_4G_LIMITED, is_support)); | ||||
| auto mem_type = is_support ? RT_MEMORY_TS_4G : RT_MEMORY_HBM; | auto mem_type = is_support ? RT_MEMORY_TS_4G : RT_MEMORY_HBM; | ||||
| // malloc args memory | // malloc args memory | ||||
| if (total_args_size_ == 0) { | |||||
| GELOGW("DavinciModel::MallocKnownArgs total_args_size_ equals to zero."); | |||||
| return SUCCESS; | |||||
| } | |||||
| rtError_t rt_ret = rtMalloc(&args_, total_args_size_, mem_type); | |||||
| if (rt_ret != RT_ERROR_NONE) { | |||||
| GELOGE(RT_FAILED, "Call rtMalloc failed, ret: 0x%X", rt_ret); | |||||
| return RT_ERROR_TO_GE_STATUS(rt_ret); | |||||
| if (total_args_size_ != 0) { | |||||
| rt_ret = rtMalloc(&args_, total_args_size_, RT_MEMORY_HBM); | |||||
| if (rt_ret != RT_ERROR_NONE) { | |||||
| REPORT_CALL_ERROR("E19999", "Call rtMalloc failed, size:%u, ret: 0x%X when DavinciModel %s", | |||||
| total_args_size_, rt_ret, __FUNCTION__); | |||||
| GELOGE(RT_FAILED, "Call rtMalloc failed, ret: 0x%X", rt_ret); | |||||
| return RT_ERROR_TO_GE_STATUS(rt_ret); | |||||
| } | |||||
| } | } | ||||
| // malloc dynamic and static hybrid memory | // malloc dynamic and static hybrid memory | ||||
| if (total_hybrid_args_size_ != 0) { | if (total_hybrid_args_size_ != 0) { | ||||