ggml : assert when using ggml_mul with non-F32 src1

This commit is contained in:
Georgi Gerganov 2023-08-16 19:19:46 +03:00
parent f3e90f27de
commit c290f3eee6
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
2 changed files with 3 additions and 1 deletions

2
ggml.c
View file

@ -9140,6 +9140,8 @@ static void ggml_compute_forward_mul(
const struct ggml_tensor * src0, const struct ggml_tensor * src0,
const struct ggml_tensor * src1, const struct ggml_tensor * src1,
struct ggml_tensor * dst) { struct ggml_tensor * dst) {
GGML_ASSERT(src1->type == GGML_TYPE_F32 && "only f32 src1 supported for now");
switch (src0->type) { switch (src0->type) {
case GGML_TYPE_F32: case GGML_TYPE_F32:
{ {