Browse Source

fix ts 4g memory bug

pull/1414/head
wxl 4 years ago
parent
commit
0b533a725b
1 changed files with 0 additions and 5 deletions
  1. +0
    -5
      ge/graph/load/model_manager/davinci_model.cc

+ 0
- 5
ge/graph/load/model_manager/davinci_model.cc View File

@@ -2958,11 +2958,6 @@ Status DavinciModel::MallocKnownArgs() {

if (total_args_size_ != 0) {
rt_ret = rtMalloc(&args_, total_args_size_, mem_type);
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__);
if (total_args_size_ != 0) {
rt_ret = rtMalloc(&args_, total_args_size_, RT_MEMORY_HBM);
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);


Loading…
Cancel
Save