colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.jax: fix MKL-enabled build on Intel CPUs

Browse Source 
Intel doesn't know how to build a CPU it turns out. Don't use them for floating point arithmetic, even with their own MKL!
This commit is contained in:
Samuel Ainsworth 2022-02-27 08:06:53 +00:00
parent 4ce387ea40
commit 069b742520
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkgs/development/python-modules/jax/default.nix
View File

@ -64,8 +64,15 @@ buildPythonPackage rec {
"tests/"
];
# See https://github.com/google/jax/issues/9705.
disabledTests = lib.optionals usingMKL [ "test_custom_root_with_aux" ];
# See
# * https://github.com/google/jax/issues/9705
# * https://discourse.nixos.org/t/getting-different-results-for-the-same-build-on-two-equally-configured-machines/17921
# * https://github.com/NixOS/nixpkgs/issues/161960
disabledTests = lib.optionals usingMKL [
"test_custom_linear_solve_cholesky"
"test_custom_root_with_aux"
"testEigvalsGrad_shape"
];
pythonImportsCheck = [
"jax"