Kaynağa Gözat

Fix RMS norm in GGML (#191)

Nebula 2 yıl önce
ebeveyn
işleme
9b4a15b17d
1 değiştirilmiş dosya ile 1 ekleme ve 0 silme
  1. 1 0
      ggml.c

+ 1 - 0
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;