]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
ggml : fix n_threads_cur initialization with one thread (#9538)
authorslaren <redacted>
Wed, 18 Sep 2024 17:13:08 +0000 (19:13 +0200)
committerGitHub <redacted>
Wed, 18 Sep 2024 17:13:08 +0000 (10:13 -0700)
* ggml : fix n_threads_cur initialization with one thread

* Update ggml/src/ggml.c

---------

Co-authored-by: Max Krasnyansky <redacted>
ggml/src/ggml.c

index bccb62377d7394c1a2fc813627019de3812f023f..c5c98dbe4bf68c7c1377f281b98043e13ebaa18a 100644 (file)
@@ -20239,6 +20239,7 @@ enum ggml_status ggml_graph_compute(struct ggml_cgraph * cgraph, struct ggml_cpl
             ggml_graph_compute_thread(&threadpool->workers[omp_get_thread_num()]);
         }
     } else {
+        atomic_store_explicit(&threadpool->n_threads_cur, 1, memory_order_relaxed);
         ggml_graph_compute_thread(&threadpool->workers[0]);
     }
 #else