p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!16043 fix If the frequency is not set on the NPU, the CompileGraph fails 

From: @zhaozhenlong
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-05-07 14:56:18 +08:00 committed by Gitee
parent dc4815860b aff228349b
commit ac5af72836
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/runtime/agent/npu/subgraph_npu_kernel.cc
View File

@ -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_);