From 9b4a15b17d8395eb075379b140fcd0b0283f4ef6 Mon Sep 17 00:00:00 2001
From: Nebula <infinitewormhole@gmail.com>
Date: Wed, 15 Mar 2023 19:29:25 -0400
Subject: Fix RMS norm in GGML (#191)

---
 ggml.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ggml.c b/ggml.c
index eee54f7..535c7b7 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;
-- 
cgit v1.2.3