From 84b58ec7f3e0c352f2a5778196999d2ae3a6f1c1 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Mon, 9 Oct 2023 10:33:12 -0700
Subject: [PATCH] Increase minimum scipy version to 1.9.

Scipy 1.9 appears to fix some crashes on Mac ARM.

PiperOrigin-RevId: 571977068
---
 jaxlib/setup.py | 7 ++++++-
 setup.py        | 3 ++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/jaxlib/setup.py b/jaxlib/setup.py
index 788f3fd4d..8449f9d5b 100644
--- a/jaxlib/setup.py
+++ b/jaxlib/setup.py
@@ -60,7 +60,12 @@ setup(
     author_email='jax-dev@google.com',
     packages=['jaxlib', 'jaxlib.xla_extension'],
     python_requires='>=3.9',
-    install_requires=['scipy>=1.7', 'numpy>=1.22', 'ml_dtypes>=0.2.0'],
+    install_requires=[
+        'scipy>=1.9',
+        "scipy>=1.11.1; python_version>='3.12'",
+        'numpy>=1.22',
+        'ml_dtypes>=0.2.0',
+    ],
     extras_require={
       'cuda11_pip': [
         "nvidia-cublas-cu11>=11.11",
diff --git a/setup.py b/setup.py
index 93e7f36e4..7f1b4bd52 100644
--- a/setup.py
+++ b/setup.py
@@ -78,7 +78,8 @@ setup(
         "numpy>=1.23.2; python_version>='3.11'",
         "numpy>=1.26.0; python_version>='3.12'",
         'opt_einsum',
-        'scipy>=1.7',
+        'scipy>=1.9',
+        "scipy>=1.11.1; python_version>='3.12'",
         # Required by xla_bridge.discover_pjrt_plugins for forwards compat with
         # Python versions < 3.10. Can be dropped when 3.10 is the minimum
         # required Python version.