Fix CUDA diag_mask_inf tests with LLAMA_FAST
This commit is contained in:
parent
f2eb19bd8b
commit
4fb1bd85a0
1 changed files with 1 additions and 1 deletions
2
ggml.c
2
ggml.c
|
@ -10928,7 +10928,7 @@ static void ggml_compute_forward_diag_mask_inf(
|
||||||
switch (src0->type) {
|
switch (src0->type) {
|
||||||
case GGML_TYPE_F32:
|
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;
|
} break;
|
||||||
default:
|
default:
|
||||||
{
|
{
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue