root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-24 10:24:35 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

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;