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

cache number invalid add

Browse Source 
Signed-off-by: w30052974 <wangyifan94@huawei.com>
This commit is contained in:
w30052974 2024-11-30 09:52:11 +08:00
parent eab4cb80d1
commit cf1eba8a1d
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;
}