Browse Source

use rtMemGetInfo instead of rtMemGetInfoEx

pull/2066/head
lichun 3 years ago
parent
commit
6861b40dda
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/manager/graph_var_manager.cc

+ 1
- 1
ge/graph/manager/graph_var_manager.cc View File

@@ -826,7 +826,7 @@ Status VarManager::GetTotalMemorySize(size_t &total_mem_size) {
return RT_FAILED;
}
size_t free_mem = 0;
rt_ret = rtMemGetInfoEx(RT_MEMORYINFO_HBM, &free_mem, &total_mem_size);
rt_ret = rtMemGetInfo(&free_mem, &total_mem_size);
if (rt_ret != RT_ERROR_NONE) {
REPORT_CALL_ERROR("E19999", "Call rtMemGetInfo failed, ret:0x%X", rt_ret);
GELOGE(RT_FAILED, "[Call][RtMemGetInfo] failed, ret:0x%X", rt_ret);


Loading…
Cancel
Save