aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNebula <infinitewormhole@gmail.com>2023-03-15 19:29:25 -0400
committerGitHub <noreply@github.com>2023-03-15 19:29:25 -0400
commit9b4a15b17d8395eb075379b140fcd0b0283f4ef6 (patch)
tree2c7ce42c531559d72b95dff33621306b55b75fe2
parent6eac39ba953acaeec396cea2969dbf413907e2ec (diff)
Fix RMS norm in GGML (#191)
-rw-r--r--ggml.c1
1 files changed, 1 insertions, 0 deletions
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;