Merge pull request #24223 from hawkinsp/ub

seberg · web-flow · commit 8f15f5eaf2cd · 2023-07-21T08:37:44.000+02:00
BUG: Fix undefined behavior in complex pow().
diff --git a/numpy/core/src/npymath/npy_math_complex.c.src b/numpy/core/src/npymath/npy_math_complex.c.src
@@ -481,7 +481,7 @@ npy_cpow@c@ (@ctype@ a, @ctype@ b)
           r = npy_cpack@c@(NPY_NAN@C@, NPY_NAN@C@);
           return r;
     }
-    if (bi == 0 && (n=(npy_intp)br) == br) {
+    if (bi == 0 && br > -100 && br < 100 && (n=(npy_intp)br) == br) {
         if (n == 1) {
             /* unroll: handle inf better */
             return npy_cpack@c@(ar, ai);
