From a6e886b87424ebe1c9abff8a6beb3cc6d08b06c3 Mon Sep 17 00:00:00 2001 From: yeyunpeng Date: Wed, 23 Dec 2020 17:28:40 +0800 Subject: [PATCH] fix check chip error --- mindspore/lite/src/runtime/agent/npu/npu_manager.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/src/runtime/agent/npu/npu_manager.cc b/mindspore/lite/src/runtime/agent/npu/npu_manager.cc index fb20549a98..0e1d802363 100644 --- a/mindspore/lite/src/runtime/agent/npu/npu_manager.cc +++ b/mindspore/lite/src/runtime/agent/npu/npu_manager.cc @@ -112,11 +112,12 @@ bool NPUManager::IsKirinChip() { if (index == string::npos) { continue; } - auto kirin_number_str = line.substr(index + 5); - if (kirin_number_str == "9000E") { + // support Kirin 990 5G\990E\9000E + if (line.find("990") != string::npos || line.find("9000") != string::npos) { cpu_info.close(); return true; } + auto kirin_number_str = line.substr(index + 5); auto kirin_number = atoi(kirin_number_str.c_str()); if (kirin_number >= 985 || kirin_number == 810 || kirin_number == 820) { cpu_info.close();