python3Packages.jax: fix build on aarch64-darwin

2023-09-29 18:47:05 -07:00
parent ed66c817ab
commit 97afe3b9ec
1 changed files with 8 additions and 0 deletions
@@ -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) [