ggml : check ggml_add src1 type (ggml/708)

Co-authored-by: Judd <[email protected]>
teleprint-me · Jan 27, 2024 · e976423 · e976423
1 parent 35a2ee9
commit e976423
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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:
             {