]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
ggml : check ggml_add src1 type (#708)
authorJudd <redacted>
Fri, 26 Jan 2024 13:04:01 +0000 (21:04 +0800)
committerGitHub <redacted>
Fri, 26 Jan 2024 13:04:01 +0000 (15:04 +0200)
Co-authored-by: Judd <redacted>
src/ggml.c

index cbf2d4bddddb83ac106d79ca4216aa8f114efffd..922be198804e72f9bf7ca4aed0e2640d6183db7d 100644 (file)
@@ -7443,7 +7443,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:
             {