From 6861b40dda6a4571d7802500ee2e46709c933e49 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Thu, 29 Jul 2021 21:55:02 +0800
Subject: [PATCH] use rtMemGetInfo instead of rtMemGetInfoEx

---
 ge/graph/manager/graph_var_manager.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/graph/manager/graph_var_manager.cc b/ge/graph/manager/graph_var_manager.cc
index 6deb1a54..9bc412cb 100755
--- a/ge/graph/manager/graph_var_manager.cc
+++ b/ge/graph/manager/graph_var_manager.cc
@@ -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);