CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-28 08:31:25 +00:00
Code Issues 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
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; 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;