Fix RMS norm in GGML (#191)

diff --git a/ggml.c b/ggml.c
index eee54f7ffc89029149567a1f522b4077621bd2f2..535c7b7d281dd3dbecaa7dcaa176f0fb11ef0736 100644 (file)
--- a/ggml.c
+++ b/ggml.c
@@ -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;