fix check chip error

pull/10395/head
yeyunpeng 4 years ago
parent 12602abd44
commit a6e886b874

@ -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); // support Kirin 990 5G\990E\9000E
if (kirin_number_str == "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