ggml : check ggml_add src1 type (ggml/708)

Co-authored-by: Judd <foldl@boxvest.com>
2024-01-26 21:04:01 +08:00
parent 4bbb60efce
commit 2d2c93a798
1 changed files with 6 additions and 1 deletions
--- a/ggml.c
+++ b/ggml.c
@@ -7443,7 +7443,12 @@ static void ggml_compute_forward_add(
    switch (src0->type) {
        case GGML_TYPE_F32:
            {
+                if (src1->type == GGML_TYPE_F32) {
                    ggml_compute_forward_add_f32(params, src0, src1, dst);
+                }
+                else {
+                    GGML_ASSERT(false);
+                }
            } break;
        case GGML_TYPE_F16:
            {