From 2d2c93a798af90a9c82b27bac78dc20746069025 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 cbf2d4b..922be19 100644 --- a/ggml.c +++ b/ggml.c @@ -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: {