vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix RMS norm in GGML

Browse source
This commit is contained in:
nebulatgs 2023-03-15 19:04:08 -04:00
parent 6eac39ba95
commit fa2aa1728a
No known key found for this signature in database
GPG key ID: 10F57782EEBBEC9A
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ggml.c
View file

@ -9314,6 +9314,7 @@ void ggml_graph_compute(struct ggml_context * ctx, struct ggml_cgraph * cgraph)
node->n_tasks = n_threads;
} break;
case GGML_OP_NORM:
case GGML_OP_RMS_NORM:
{
node->n_tasks = n_threads;
} break;