From e9764230054e01553bdead6f2bfd8e001869599d Mon Sep 17 00:00:00 2001 From: Judd Date: Fri, 26 Jan 2024 21:04:01 +0800 Subject: [PATCH] ggml : check ggml_add src1 type (ggml/708) Co-authored-by: Judd --- ggml.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ggml.c b/ggml.c index 8f57003e0..11a3114e7 100644 --- 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: {