Browse Source

!1770 MallocExMem

From: @dimitri_rose
Reviewed-by: @tangqunzhang,@wqtshg
Signed-off-by: @wqtshg
tags/v1.3.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
5fa53be481
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/graph/load/model_manager/davinci_model.cc

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

@@ -4096,7 +4096,7 @@ uint8_t *DavinciModel::MallocFeatureMapMem(size_t data_size) {
Status DavinciModel::MallocExMem() { Status DavinciModel::MallocExMem() {
char ge_static_mem_env[MMPA_MAX_PATH] = {0x00}; char ge_static_mem_env[MMPA_MAX_PATH] = {0x00};
INT32 res_static_memory = mmGetEnv(kEnvGeuseStaticMemory, ge_static_mem_env, MMPA_MAX_PATH); INT32 res_static_memory = mmGetEnv(kEnvGeuseStaticMemory, ge_static_mem_env, MMPA_MAX_PATH);
for (auto it : runtime_param_.memory_infos) {
for (auto &it : runtime_param_.memory_infos) {
auto mem_size = it.second.memory_size; auto mem_size = it.second.memory_size;
if (mem_size == 0) { if (mem_size == 0) {
continue; continue;
@@ -4167,7 +4167,7 @@ void DavinciModel::FreeFeatureMapMem() {
void DavinciModel::FreeExMem() { void DavinciModel::FreeExMem() {
char ge_static_mem_env[MMPA_MAX_PATH] = {0x00}; char ge_static_mem_env[MMPA_MAX_PATH] = {0x00};
INT32 res_static_memory = mmGetEnv(kEnvGeuseStaticMemory, ge_static_mem_env, MMPA_MAX_PATH); INT32 res_static_memory = mmGetEnv(kEnvGeuseStaticMemory, ge_static_mem_env, MMPA_MAX_PATH);
for (auto it : runtime_param_.memory_infos) {
for (auto &it : runtime_param_.memory_infos) {
// free when session destory // free when session destory
if ((kSessionScopeMemory & it.first) == kSessionScopeMemory) { if ((kSessionScopeMemory & it.first) == kSessionScopeMemory) {
continue; continue;


Loading…
Cancel
Save