zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

JAX: add v0.4.28 (#44112)

Browse Source
This commit is contained in:
Adam J. Stewart
2024-05-17 21:58:44 +02:00
committed by GitHub
parent 6be28aa303
commit 8d01e8c978
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
var/spack/repos/builtin/packages/py-jax/package.py
View File

@@ -24,6 +24,7 @@ class PyJax(PythonPackage):
license("Apache-2.0")
maintainers("adamjstewart", "jonas-eschle")
version("0.4.28", sha256="dcf0a44aff2e1713f0a2b369281cd5b79d8c18fc1018905c4125897cb06b37e9")
version("0.4.27", sha256="f3d7f19bdc0a17ccdb305086099a5a90c704f904d4272a70debe06ae6552998c")
version("0.4.26", sha256="2cce025d0a279ec630d550524749bc8efe25d2ff47240d2a7d4cfbc5090c5383")
version("0.4.25", sha256="a8ee189c782de2b7b2ffb64a8916da380b882a617e2769aa429b71d79747b982")
@@ -83,6 +84,7 @@ class PyJax(PythonPackage):
# jax/_src/lib/__init__.py
# https://github.com/google/jax/commit/8be057de1f50756fe7522f7e98b2f30fad56f7e4
for v in [
"0.4.28",
"0.4.27",
"0.4.26",
"0.4.25",
@@ -113,6 +115,7 @@ class PyJax(PythonPackage):
depends_on(f"py-jaxlib@:{v}", when=f"@{v}")
# See _minimum_jaxlib_version in jax/version.py
depends_on("py-jaxlib@0.4.27:", when="@0.4.28:")
depends_on("py-jaxlib@0.4.23:", when="@0.4.27:")
depends_on("py-jaxlib@0.4.20:", when="@0.4.25:")
depends_on("py-jaxlib@0.4.19:", when="@0.4.21:")

4
var/spack/repos/builtin/packages/py-jaxlib/package.py
View File

@@ -20,6 +20,7 @@ class PyJaxlib(PythonPackage, CudaPackage):
license("Apache-2.0")
maintainers("adamjstewart")
version("0.4.28", sha256="4dd11577d4ba5a095fbc35258ddd4e4c020829ed6e6afd498c9e38ccbcdfe20b")
version("0.4.27", sha256="c2c82cd9ad3b395d5cbc0affa26a2938e52677a69ca8f0b9ef9922a52cac4f0c")
version("0.4.26", sha256="ddc14da1eaa34f23430d40ad9b9585088575cac439a2fa1c6833a247e1b221fd")
version("0.4.25", sha256="fc1197c401924942eb14185a61688d0c476e3e81ff71f9dc95e620b57c06eec8")
@@ -69,7 +70,8 @@ class PyJaxlib(PythonPackage, CudaPackage):
with default_args(type="build"):
# .bazelversion
depends_on("bazel@6.1.2", when="@0.4.11:")
depends_on("bazel@6.5.0", when="@0.4.28:")
depends_on("bazel@6.1.2", when="@0.4.11:0.4.27")
depends_on("bazel@5.1.1", when="@0.3.7:0.4.10")
depends_on("bazel@5.1.0", when="@0.3.5")
depends_on("bazel@5.0.0", when="@0.3.0:0.3.2")