Browse Source

fix mem leak

tags/v1.3.0
wuweikang 4 years ago
parent
commit
e51ffe2f54
1 changed files with 8 additions and 7 deletions
  1. +8
    -7
      ge/graph/load/model_manager/model_manager.cc

+ 8
- 7
ge/graph/load/model_manager/model_manager.cc View File

@@ -1397,6 +1397,14 @@ Status ModelManager::LaunchKernelCustAicpuSo(const string &kernel_name) {
vector<void *> allocated_mem; vector<void *> allocated_mem;
rtError_t status; rtError_t status;
rtStream_t stream = nullptr; rtStream_t stream = nullptr;
std::function<void()> callback = [&]() {
for (auto mem : allocated_mem) {
GE_CHK_RT(rtFree(mem));
}
GE_CHK_RT(rtStreamDestroy(stream));
};
GE_MAKE_GUARD(release, callback);

vector<CustAicpuSoBuf> v_cust_so; vector<CustAicpuSoBuf> v_cust_so;
void *args = nullptr; void *args = nullptr;


@@ -1471,13 +1479,6 @@ Status ModelManager::LaunchKernelCustAicpuSo(const string &kernel_name) {
GELOGE(RT_FAILED, "[Call][RtStreamSynchronize] fail, ret = 0x%X", status); GELOGE(RT_FAILED, "[Call][RtStreamSynchronize] fail, ret = 0x%X", status);
return RT_ERROR_TO_GE_STATUS(status); return RT_ERROR_TO_GE_STATUS(status);
} }
std::function<void()> callback = [&]() {
for (auto mem : allocated_mem) {
GE_CHK_RT(rtFree(mem));
}
GE_CHK_RT(rtStreamDestroy(stream));
};
GE_MAKE_GUARD(release, callback);
GELOGI("Cpu kernel launch task success."); GELOGI("Cpu kernel launch task success.");
return SUCCESS; return SUCCESS;
} }


Loading…
Cancel
Save