Avoid division-by-zero on 0-weights

This commit is contained in:
Sigbjørn Skjæret 2024-06-07 23:57:04 +02:00 committed by GitHub
parent da799b4189
commit 0225cfbfb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1754,7 +1754,7 @@ static float make_q3_quants(int n, int nmax, const float * restrict x, int8_t *
for (int i = 0; i < n; ++i) {
L[i] += nmax;
}
return sumlx / suml2;
return suml2 ? sumlx / suml2 : 0.0f;
}
for (int i = 0; i < n; ++i) {
int l = nearest_int(iscale * x[i]);
@ -2158,7 +2158,7 @@ static float make_qp_quants(int n, int nmax, const float * restrict x, uint8_t *
break;
}
}
return sumlx/suml2;
return suml2 ? sumlx/suml2 : 0.0f;
}
static void quantize_row_q2_K_impl(const float * restrict x, block_q2_K * restrict y, int k, const float * restrict quant_weights) {