ggml : add GGML_DEFAULT_N_THREADS

pull/545/head master-a3a2a0e
Georgi Gerganov 1 year ago
parent d990e3fffc
commit a3a2a0eda8
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -9363,7 +9363,7 @@ struct ggml_cgraph ggml_build_forward(struct ggml_tensor * tensor) {
struct ggml_cgraph result = {
/*.n_nodes =*/ 0,
/*.n_leafs =*/ 0,
/*.n_threads =*/ 0,
/*.n_threads =*/ GGML_DEFAULT_N_THREADS,
/*.work_size =*/ 0,
/*.work =*/ NULL,
/*.nodes =*/ { NULL },

@ -182,6 +182,7 @@ extern "C" {
#define GGML_MAX_PARAMS 16
#define GGML_MAX_CONTEXTS 64
#define GGML_MAX_OPT 4
#define GGML_DEFAULT_N_THREADS 4
#ifdef __ARM_NEON
// we use the built-in 16-bit float type

Loading…
Cancel
Save