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

2024-09-05 02:19:39 +02:00 · 2024-09-05 02:19:39 +02:00 · bdf314f38a
commit bdf314f38a
parent 581c305186
1 changed files with 3 additions and 0 deletions
--- 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