openharmony/ai_neural_network_runtime
1
0
Fork 0
mirror of https://gitee.com/openharmony/ai_neural_network_runtime synced 2024-12-04 05:50:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

!222 nnrt memory leak fixed 2.0

Browse Source 
Merge pull request !222 from wangyifan/master
This commit is contained in:
openharmony_ci 2024-09-13 02:15:16 +00:00 committed by Gitee
commit 1b26d4aa9c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frameworks/native/neural_network_core/nnrt_client.cpp
View File

@ -48,11 +48,13 @@ NNRtServiceApi& NNRtServiceApi::GetInstance()
// Assumes there can be multiple instances of NN API
std::string nnrtLibraryName = "libdllite_service_client.z.so";
libNNRtService = dlopen(nnrtLibraryName.c_str(), RTLD_LAZY | RTLD_NODELETE);
if (libNNRtService == nullptr) {
LOGE("LoadNNRtService error: unable to open library %{public}s", nnrtLibraryName.c_str());
nnrtService.m_serviceAvailable = false;
return nnrtService;
libNNRtService = dlopen(nnrtLibraryName.c_str(), RTLD_LAZY | RTLD_NODELETE);
if (libNNRtService == nullptr) {
LOGE("LoadNNRtService error: unable to open library %{public}s", nnrtLibraryName.c_str());
nnrtService.m_serviceAvailable = false;
return nnrtService;
}
}
LoadFunction(libNNRtService, "CheckModelSizeFromPath", &nnrtService.CheckModelSizeFromPath);