diff --git a/mindspore/lite/src/runtime/agent/npu/subgraph_npu_kernel.cc b/mindspore/lite/src/runtime/agent/npu/subgraph_npu_kernel.cc index 15a3da9138..6e43f3fb66 100644 --- a/mindspore/lite/src/runtime/agent/npu/subgraph_npu_kernel.cc +++ b/mindspore/lite/src/runtime/agent/npu/subgraph_npu_kernel.cc @@ -206,8 +206,13 @@ int SubGraphNpuKernel::Init() { MS_ASSERT(npu_manager_ != nullptr); - npu_manager_->AddModel(model_buffer_data, GetOMModelName(), - static_cast(context_)->GetNpuInfo().frequency_); + int frequency = static_cast(context_)->GetNpuInfo().frequency_; + if (frequency != hiai::AiModelDescription_Frequency_LOW && frequency != hiai::AiModelDescription_Frequency_MEDIUM && + frequency != hiai::AiModelDescription_Frequency_HIGH && + frequency != hiai::AiModelDescription_Frequency_EXTREME) { + frequency = hiai::AiModelDescription_Frequency_HIGH; + } + npu_manager_->AddModel(model_buffer_data, GetOMModelName(), frequency); executor_ = new (std::nothrow) mindspore::lite::NPUExecutor(GetOMModelName(), npu_manager_);