From 069b7425200b747ec6516a0663d646b9fb3927eb Mon Sep 17 00:00:00 2001 From: Samuel Ainsworth Date: Sun, 27 Feb 2022 08:06:53 +0000 Subject: [PATCH] python3Packages.jax: fix MKL-enabled build on Intel CPUs 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! --- pkgs/development/python-modules/jax/default.nix | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/jax/default.nix b/pkgs/development/python-modules/jax/default.nix index a616a83255b6..d5e536540195 100644 --- a/pkgs/development/python-modules/jax/default.nix +++ b/pkgs/development/python-modules/jax/default.nix @@ -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"