Skip test_layout_donation_mismatching_in_and_out_fails on CPU and GPU

PiperOrigin-RevId: 668652286
jax-ml · Aug 28, 2024 · 12642d3 · 12642d3
1 parent 59d75b3
commit 12642d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/layout_test.py b/tests/layout_test.py
@@ -575,6 +575,7 @@ def f(x):
     f(arr)
     self.assertTrue(arr.is_deleted())
 
+  @jtu.skip_on_devices('cpu', 'gpu')
   def test_layout_donation_mismatching_in_and_out_fails(self):
     mesh = jtu.create_global_mesh((2, 2), ('x', 'y'))
     s = NamedSharding(mesh, P('x', 'y'))