ggml : fix another case of quants nans (#7387)

This commit is contained in:
slaren 2024-05-19 17:08:46 +02:00 committed by GitHub
parent 5ca49cbecd
commit e4e6f67be6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1149,7 +1149,7 @@ static float make_qx_quants(int n, int nmax, const float * restrict x, int8_t *
sumlx += w*x[i]*l; sumlx += w*x[i]*l;
suml2 += w*l*l; suml2 += w*l*l;
} }
float scale = sumlx/suml2; float scale = suml2 ? sumlx/suml2 : 0.0f;
if (return_early) return suml2 > 0 ? 0.5f*(scale + 1/iscale) : 1/iscale; if (return_early) return suml2 > 0 ? 0.5f*(scale + 1/iscale) : 1/iscale;
float best = scale * sumlx; float best = scale * sumlx;
for (int is = -9; is <= 9; ++is) { for (int is = -9; is <= 9; ++is) {