From f8bd998b4e6926445ab0fd7c95e1e29687c8edcf Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Mon, 10 Feb 2025 17:52:33 +0100 Subject: [PATCH] python3Packages.jax-cuda12-pjrt: don't try to build on aarch64-linux --- .../python-modules/jax-cuda12-pjrt/default.nix | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix b/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix index b22e6fc6350f..bb0a01b2af4f 100644 --- a/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix +++ b/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix @@ -29,16 +29,16 @@ let # When upgrading, you can get these hashes from jaxlib/prefetch.sh. See # https://github.com/google/jax/issues/12879 as to why this specific URL is the correct index. - # upstream does not distribute jax-cuda12-pjrt 0.4.38 binaries for aarch64-linux + # upstream does not distribute jax-cuda12-pjrt binaries for aarch64-linux srcs = { "x86_64-linux" = fetchurl { url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_x86_64.whl"; hash = "sha256-0jgzwbiF2WwnZAAOlQUvK1gnx31JLqaPZ+kDoTJlbbs="; }; - "aarch64-linux" = fetchurl { - url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_aarch64.whl"; - hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA="; - }; + # "aarch64-linux" = fetchurl { + # url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_aarch64.whl"; + # hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA="; + # }; }; in buildPythonPackage {