Update zscal.c

OpenMathLib · Jul 31, 2024 · 37e3c5f · 37e3c5f
1 parent 7caa356
commit 37e3c5f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kernel/zarch/zscal.c b/kernel/zarch/zscal.c
@@ -381,7 +381,8 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r, FLOAT da_i,
         	x[i + 1] = da_i * x[i];
 	else
 		x[i + 1] = NAN;
-        x[i] = temp0;
+	if (isfinite(x[i]))
+          x[i] = temp0;
         i += 2;
         j++;
 
@@ -396,6 +397,7 @@ int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r, FLOAT da_i,
       while (j < n) {
 
         temp0 = da_r * x[i];
+	if (isinf(x[i])
         x[i + 1] = da_r * x[i + 1];
         x[i] = temp0;
         i += 2;