aditya/llama.cpp
1
0
Fork 0
mirror of https://git.adityakumar.xyz/llama.cpp.git synced 2024-11-09 23:29:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix RMS norm in GGML (#191)

Browse source
This commit is contained in:
Nebula 2023-03-15 19:29:25 -04:00 committed by GitHub
parent 6eac39ba95
commit 9b4a15b17d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show stats 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; node->n_tasks = n_threads;
} break; } break;
case GGML_OP_NORM: case GGML_OP_NORM:
case GGML_OP_RMS_NORM:
{ {
node->n_tasks = n_threads; node->n_tasks = n_threads;
} break; } break;