Commit 97afe3b9 authored by Samuel Ainsworth's avatar Samuel Ainsworth
Browse files

python3Packages.jax: fix build on aarch64-darwin

parent ed66c817

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) [