Browse Source

fix check chip error

tags/v1.1.0
yeyunpeng 5 years ago
parent
commit
a6e886b874
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      mindspore/lite/src/runtime/agent/npu/npu_manager.cc

+ 3
- 2
mindspore/lite/src/runtime/agent/npu/npu_manager.cc View File

@@ -112,11 +112,12 @@ bool NPUManager::IsKirinChip() {
if (index == string::npos) { if (index == string::npos) {
continue; 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(); cpu_info.close();
return true; return true;
} }
auto kirin_number_str = line.substr(index + 5);
auto kirin_number = atoi(kirin_number_str.c_str()); auto kirin_number = atoi(kirin_number_str.c_str());
if (kirin_number >= 985 || kirin_number == 810 || kirin_number == 820) { if (kirin_number >= 985 || kirin_number == 810 || kirin_number == 820) {
cpu_info.close(); cpu_info.close();


Loading…
Cancel
Save