mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increase minimum scipy version to 1.9.

Browse Source 
Scipy 1.9 appears to fix some crashes on Mac ARM.

PiperOrigin-RevId: 571977068
This commit is contained in:
Peter Hawkins 2023-10-09 10:33:12 -07:00 committed by jax authors
parent 1ec51f4cba
commit 84b58ec7f3
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
jaxlib/setup.py
View File

@ -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",

3
setup.py
View File

@ -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.