diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2024-05-17 21:58:44 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-17 21:58:44 +0200 |
| commit | 8d01e8c978169d2da972073bd4ad60942e016595 (patch) | |
| tree | d68c7b3ac58ed4f8c5b37b9c13034f3f94d29814 /var | |
| parent | 6be28aa303548e936d9bdcae2edb6c5f91183010 (diff) | |
| download | spack-8d01e8c978169d2da972073bd4ad60942e016595.tar.gz spack-8d01e8c978169d2da972073bd4ad60942e016595.tar.bz2 spack-8d01e8c978169d2da972073bd4ad60942e016595.tar.xz spack-8d01e8c978169d2da972073bd4ad60942e016595.zip | |
JAX: add v0.4.28 (#44112)
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-jax/package.py | 3 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/py-jaxlib/package.py | 4 |
2 files changed, 6 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-jax/package.py b/var/spack/repos/builtin/packages/py-jax/package.py index 2394c0b9ae..27e46b451f 100644 --- a/var/spack/repos/builtin/packages/py-jax/package.py +++ b/var/spack/repos/builtin/packages/py-jax/package.py @@ -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:") diff --git a/var/spack/repos/builtin/packages/py-jaxlib/package.py b/var/spack/repos/builtin/packages/py-jaxlib/package.py index fcca93ad04..4904d07ad1 100644 --- a/var/spack/repos/builtin/packages/py-jaxlib/package.py +++ b/var/spack/repos/builtin/packages/py-jaxlib/package.py @@ -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") |