From aff228349be035753ab50648f5d387c09c9fce67 Mon Sep 17 00:00:00 2001
From: zhaozhenlong <zhaozhenlong1@huawei.com>
Date: Fri, 7 May 2021 11:29:14 +0800
Subject: [PATCH] fix npu default frequency

---
 .../lite/src/runtime/agent/npu/subgraph_npu_kernel.cc    | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

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<const lite::InnerContext *>(context_)->GetNpuInfo().frequency_);
+    int frequency = static_cast<const lite::InnerContext *>(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_);