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

fix cache model size

Browse Source 
Signed-off-by: w30052974 <wangyifan94@huawei.com>
This commit is contained in:
w30052974 2024-10-18 11:29:30 +08:00
parent b734414d46
commit 527b02827c
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frameworks/native/neural_network_core/neural_network_core.cpp
View File

@ -755,7 +755,7 @@ NNRT_API OH_NN_ReturnCode OH_NNCompilation_Build(OH_NNCompilation *compilation)
configContents.push_back('0');
}
configContents["isExceedRamLimit"] = configContents;
configs["isExceedRamLimit"] = configContents;
compilationImpl->compiler->SetExtensionConfig(configs);
bool isBuild = compilationImpl->compiler->IsBuild();

2
frameworks/native/neural_network_runtime/nncompiler.cpp
View File

@ -728,7 +728,7 @@ OH_NN_ReturnCode NNCompiler::SetExtensionConfig(const std::unordered_map<std::st
m_extensionConfig.isNpuFmShared = true;
LOGI("[NNCompiler] SetExtensionConfig NpuFmShared enabled.");
}
if (cofigs.find(EXTENSION_KEY_IS_EXCEED_RAMLIMIT) != configs.end()) {
if (configs.find(EXTENSION_KEY_IS_EXCEED_RAMLIMIT) != configs.end()) {
std::vector<char> value = configs.at(EXTENSION_KEY_IS_EXCEED_RAMLIMIT);
if (value.empty()) {
LOGE("[NNCompiler] SetExtensionConfig get empty model name from configs");