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

!242 cache number invalid add

Browse Source 
Merge pull request !242 from wangyifan/master
This commit is contained in:
openharmony_ci 2024-11-30 03:43:50 +00:00 committed by Gitee
commit ea42fc43ce
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frameworks/native/neural_network_runtime/neural_network_runtime.cpp
View File

@ -596,8 +596,8 @@ NNRT_API bool OH_NNModel_HasCache(const char *cacheDir, const char *modelName, u
return false;
}
if (fileNumber > FILE_NUMBER_MAX) {
LOGE("OH_NNModel_HasCache fileNumber is more than 100");
if (fileNumber <= 0 || fileNumber > FILE_NUMBER_MAX) {
LOGE("OH_NNModel_HasCache fileNumber is invalid or more than 100");
std::filesystem::remove_all(cacheInfoPath);
return false;
}