alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

Merge pull request #233579 from wegank/python-jax-fix

Browse source 
python310Packages.jax: fix build on aarch64-linux
This commit is contained in:
Weijia Wang 2023-05-23 12:52:30 +03:00 committed by GitHub
parent 25e66dcaa5 12f7d77929
commit 6a57fe07ec
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkgs/development/python-modules/jax/default.nix
View file

@ -89,7 +89,7 @@ buildPythonPackage rec {
"test_custom_linear_solve_cholesky"
"test_custom_root_with_aux"
"testEigvalsGrad_shape"
] ++ lib.optionals (stdenv.isAarch64 && stdenv.isDarwin) [
] ++ lib.optionals stdenv.isAarch64 [
# See https://github.com/google/jax/issues/14793.
"test_for_loop_fixpoint_correctly_identifies_loop_varying_residuals_unrolled_for_loop"
"testQdwhWithRandomMatrix3"
@ -107,6 +107,9 @@ buildPythonPackage rec {
"tests/nn_test.py"
"tests/random_test.py"
"tests/sparse_test.py"
] ++ lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [
# RuntimeWarning: invalid value encountered in cast
"tests/lax_test.py"
];
# As of 0.3.22, `import jax` does not work without jaxlib being installed.