python310Packages.jax: fix build on aarch64-linux
This commit is contained in:
parent
fdd73b14e8
commit
12f7d77929
|
@ -89,7 +89,7 @@ buildPythonPackage rec {
|
||||||
"test_custom_linear_solve_cholesky"
|
"test_custom_linear_solve_cholesky"
|
||||||
"test_custom_root_with_aux"
|
"test_custom_root_with_aux"
|
||||||
"testEigvalsGrad_shape"
|
"testEigvalsGrad_shape"
|
||||||
] ++ lib.optionals (stdenv.isAarch64 && stdenv.isDarwin) [
|
] ++ lib.optionals stdenv.isAarch64 [
|
||||||
# See https://github.com/google/jax/issues/14793.
|
# See https://github.com/google/jax/issues/14793.
|
||||||
"test_for_loop_fixpoint_correctly_identifies_loop_varying_residuals_unrolled_for_loop"
|
"test_for_loop_fixpoint_correctly_identifies_loop_varying_residuals_unrolled_for_loop"
|
||||||
"testQdwhWithRandomMatrix3"
|
"testQdwhWithRandomMatrix3"
|
||||||
|
@ -107,6 +107,9 @@ buildPythonPackage rec {
|
||||||
"tests/nn_test.py"
|
"tests/nn_test.py"
|
||||||
"tests/random_test.py"
|
"tests/random_test.py"
|
||||||
"tests/sparse_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.
|
# As of 0.3.22, `import jax` does not work without jaxlib being installed.
|
||||||
|
|
Loading…
Reference in a new issue