Fix CUDA diag_mask_inf tests with LLAMA_FAST

This commit is contained in:
JohannesGaessler 2024-01-03 11:20:28 +01:00
parent f2eb19bd8b
commit 4fb1bd85a0

2
ggml.c
View file

@ -10928,7 +10928,7 @@ static void ggml_compute_forward_diag_mask_inf(
switch (src0->type) {
case GGML_TYPE_F32:
{
ggml_compute_forward_diag_mask_f32(params, src0, dst, -INFINITY);
ggml_compute_forward_diag_mask_f32(params, src0, dst, -FLT_MAX);
} break;
default:
{