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

OpenBMB · Sep 5, 2024 · bdf314f · bdf314f
1 parent 581c305
commit bdf314f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/llama-bench/llama-bench.cpp b/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