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

!241 cache number limit

Browse Source 
Merge pull request !241 from wangyifan/master
This commit is contained in:
openharmony_ci 2024-11-29 03:08:53 +00:00 committed by Gitee
commit 6bbd837d05
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,6 +27,7 @@
#include <cstring>
#include <fstream>
#include <filesystem>
#include <sys/stat.h>
#include <unistd.h>
@ -46,6 +47,7 @@ const std::string NULL_HARDWARE_NAME = "default";
const std::string HARDWARE_NAME = "const.ai.nnrt_deivce";
const std::string HARDWARE_VERSION = "v5_0";
constexpr size_t HARDWARE_NAME_MAX_LENGTH = 128;
constexpr size_t FILE_NUMBER_MAX = 100; // 限制cache文件数量最大为100
NNRT_API NN_QuantParam *OH_NNQuantParam_Create()
{
@ -594,11 +596,22 @@ 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");
std::filesystem::remove_all(cacheInfoPath);
return false;
}
// determine whether cache model files exist
for (int64_t i = 0; i < fileNumber; ++i) {
std::string cacheModelPath =
std::string(cacheDir) + "/" + std::string(modelName) + std::to_string(i) + ".nncache";
exist = (exist && (stat(cacheModelPath.c_str(), &buffer) == 0));
if (!exist) {
LOGE("OH_NNModel_HasCache cacheModelPath is not existed.");
std::filesystem::remove_all(cacheInfoPath);
return false;
}
}
if (cacheVersion != version) {