From: Nebula <redacted>
Date: Wed, 15 Mar 2023 23:29:25 +0000 (-0400)
Subject: Fix RMS norm in GGML (#191)
X-Git-Tag: gguf-v0.4.0~1234
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=9b4a15b17d8395eb075379b140fcd0b0283f4ef6;p=pkg%2Fggml%2Fsources%2Fllama.cpp

Fix RMS norm in GGML (#191)
---

diff --git a/ggml.c b/ggml.c
index eee54f7f..535c7b7d 100644
--- 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;