pyrox.dev
python3Packages.jax-cuda12-pjrt: don't try to build on aarch64-linux (#380936)
authored by Pavol Rusnak and committed by GitHub f82abdfb 94a21d83
+5
-5
pkgs/development/python-modules/jax-cuda12-pjrt/default.nix
+5
-5
pkgs/development/python-modules/jax-cuda12-pjrt/default.nix
···
29
# When upgrading, you can get these hashes from jaxlib/prefetch.sh. See
30
# https://github.com/google/jax/issues/12879 as to why this specific URL is the correct index.
31
32
-
# upstream does not distribute jax-cuda12-pjrt 0.4.38 binaries for aarch64-linux
33
srcs = {
34
"x86_64-linux" = fetchurl {
35
url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_x86_64.whl";
36
hash = "sha256-0jgzwbiF2WwnZAAOlQUvK1gnx31JLqaPZ+kDoTJlbbs=";
37
};
38
-
"aarch64-linux" = fetchurl {
39
-
url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_aarch64.whl";
40
-
hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
41
-
};
42
};
43
in
44
buildPythonPackage {
···
29
# When upgrading, you can get these hashes from jaxlib/prefetch.sh. See
30
# https://github.com/google/jax/issues/12879 as to why this specific URL is the correct index.
31
32
+
# upstream does not distribute jax-cuda12-pjrt binaries for aarch64-linux
33
srcs = {
34
"x86_64-linux" = fetchurl {
35
url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_x86_64.whl";
36
hash = "sha256-0jgzwbiF2WwnZAAOlQUvK1gnx31JLqaPZ+kDoTJlbbs=";
37
};
38
+
# "aarch64-linux" = fetchurl {
39
+
# url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_aarch64.whl";
40
+
# hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
41
+
# };
42
};
43
in
44
buildPythonPackage {