From 96aa9653ce67b896064cdd3273ab18c3ef80d6d4 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Tue, 13 Jul 2021 20:59:28 +0800
Subject: [PATCH] assign graph memory max size and variable memory max size
 adaptively

---
 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 94ea5724..5138a0f5 100755
--- a/ge/graph/manager/graph_var_manager.cc
+++ b/ge/graph/manager/graph_var_manager.cc
@@ -777,7 +777,7 @@ Status VarManager::GetTotalMemorySize(size_t &total_mem_size) {
     return RT_FAILED;
   }
   size_t free_mem = 0;
-  rt_ret = rtMemGetInfo(&free_mem, &total_mem_size);
+  rt_ret = rtMemGetInfoEx(RT_MEMORYINFO_HBM, &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);