Unverified Commit b9b6068e authored by Weijia Wang's avatar Weijia Wang Committed by GitHub
Browse files

Merge pull request #258133 from samuela/samuela/jax 

python3Packages.jax: fix build on aarch64-darwin
parents f33abce1 97afe3b9

pkgs/development/python-modules/jax/default.nix

+8 −0
Original line number Diff line number Diff line
@@ -94,6 +94,14 @@ buildPythonPackage rec { 
    "test_for_loop_fixpoint_correctly_identifies_loop_varying_residuals_unrolled_for_loop"

    "testQdwhWithRandomMatrix3"

    "testScanGrad_jit_scan"



    # See https://github.com/google/jax/issues/17867.

    "test_array"

    "test_async"

    "test_copy0"

    "test_device_put"

    "test_make_array_from_callback"

    "test_make_array_from_single_device_arrays"

  ];



  disabledTestPaths = lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [