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

dsx1986 · Oct 29, 2024 · f230651 · f230651
1 parent cac8909
commit f230651
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