ggml : check ggml_add src1 type (ggml/708)

Co-authored-by: Judd <foldl@boxvest.com>
This commit is contained in:
Judd
2024-01-27 16:59:00 +02:00
committed by Georgi Gerganov
co-authored by Judd
parent 35a2ee9143
commit e976423005
+6 -1
View File
@@ -7498,7 +7498,12 @@ static void ggml_compute_forward_add(
switch (src0->type) { switch (src0->type) {
case GGML_TYPE_F32: 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; } break;
case GGML_TYPE_F16: case GGML_TYPE_F16:
{ {