diff --git a/ggml.c b/ggml.c index eee54f7ff..535c7b7d2 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;