python311Packages.jaxlib: limit platforms

to the currently supported ones, so that the top-level jaxlib can fallback to jaxlib-bin
This commit is contained in:
Someone Serge 2024-07-03 14:48:51 +00:00
parent 0ec90914b7
commit cf4bfbb264

View File

@ -72,12 +72,13 @@ let
homepage = "https://github.com/google/jax";
license = licenses.asl20;
maintainers = with maintainers; [ ndl ];
platforms = platforms.unix;
# Make this platforms.unix once Darwin is supported.
# The top-level jaxlib now falls back to jaxlib-bin on unsupported platforms.
# aarch64-darwin is broken because of https://github.com/bazelbuild/rules_cc/pull/136
# however even with that fix applied, it doesn't work for everyone:
# https://github.com/NixOS/nixpkgs/pull/184395#issuecomment-1207287129
# NOTE: We always build with NCCL; if it is unsupported, then our build is broken.
broken = effectiveStdenv.isDarwin || nccl.meta.unsupported;
platforms = platforms.linux;
};
# These are necessary at build time and run time.