]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
ggml : check ggml_add src1 type (ggml/708)
authorJudd <redacted>
Fri, 26 Jan 2024 13:04:01 +0000 (21:04 +0800)
committerGeorgi Gerganov <redacted>
Sat, 27 Jan 2024 14:59:00 +0000 (16:59 +0200)
Co-authored-by: Judd <redacted>
ggml.c

diff --git a/ggml.c b/ggml.c
index 8f57003e0ac9c6707100b41c995594f9f1f3eab7..11a3114e7f78161549e4e392662dede412ab4b92 100644 (file)
--- a/ggml.c
+++ b/ggml.c
@@ -7498,7 +7498,12 @@ static void ggml_compute_forward_add(
     switch (src0->type) {
         case GGML_TYPE_F32:
             {
-                ggml_compute_forward_add_f32(params, src0, src1, dst);
+                if (src1->type == GGML_TYPE_F32) {
+                    ggml_compute_forward_add_f32(params, src0, src1, dst);
+                }
+                else {
+                    GGML_ASSERT(false);
+                }
             } break;
         case GGML_TYPE_F16:
             {