Просмотр исходного кода

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

slaren 1 год назад
Родитель
Сommit
e4e6f67be6
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      ggml-quants.c

+ 1 - 1
ggml-quants.c

@@ -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) {