Fix overflow warning on apple silicon

1 files changed, 15 insertions, 0 deletions

diff --git a/numpy/core/src/umath/loops.c.src b/numpy/core/src/umath/loops.c.src
index 570b3ec04..bd8c78d2d 100644
--- a/numpy/core/src/umath/loops.c.src
+++ b/numpy/core/src/umath/loops.c.src
@@ -2526,6 +2526,21 @@ NPY_NO_EXPORT void
         const @ftype@ in1i = ((@ftype@ *)ip1)[1];
         const @ftype@ in2r = ((@ftype@ *)ip2)[0];
         const @ftype@ in2i = ((@ftype@ *)ip2)[1];
+#if defined(__APPLE__) && defined(__aarch64__)
+        // On macos-arm64, if this block of code was not there,
+        // when branch prediction goes wrong, the floating point exception
+        // register does not get cleared and an exception for the
+        // wrong branch is thrown.
+        if (in2i == 0) {
+            ((@ftype@ *)op1)[0] = npy_floor@c@(in1r/in2r);
+            ((@ftype@ *)op1)[1] = 0;
+        }
+        else if (in2r == 0) {
+            ((@ftype@ *)op1)[0] = npy_floor@c@(in1i/in2i);
+            ((@ftype@ *)op1)[1] = 0;
+        }
+        else
+#endif
         if (npy_fabs@c@(in2r) >= npy_fabs@c@(in2i)) {
             const @ftype@ rat = in2i/in2r;
             ((@ftype@ *)op1)[0] = npy_floor@c@((in1r + in1i*rat)/(in2r + in2i*rat));