llama-bench : fix NUL terminators in CPU name

2024-09-04 17:03:33 +02:00 · 2024-09-04 17:03:33 +02:00 · bcaa271893
commit bcaa271893
parent c8671ae282
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