Просмотр исходного кода

llama-bench : fix NUL terminators in CPU name (#9313)

slaren 1 год назад
Родитель
Сommit
bdf314f38a
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      examples/llama-bench/llama-bench.cpp

+ 3 - 0
examples/llama-bench/llama-bench.cpp

@@ -124,6 +124,9 @@ static std::string get_cpu_info() {
                         (LPBYTE)cpu_brand,
                         &cpu_brand_size) == ERROR_SUCCESS) {
         id.assign(cpu_brand, cpu_brand_size);
+        if (id.find('\0') != std::string::npos) {
+            id.resize(id.find('\0'));
+        }
     }
     RegCloseKey(hKey);
 #endif