Skip to content

Commit 8bbd738

Browse files
tomnatan30Google-ML-Automation
authored andcommitted
[JAX Shardy] #sdy Unskip another test that is now passing
PiperOrigin-RevId: 738814411
1 parent c098b36 commit 8bbd738

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/memories_test.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -756,9 +756,6 @@ def init():
756756
def test_compute_no_inputs_host_replicated(self):
757757
if xb.backend_xla_version() is not None and xb.backend_xla_version() < 3:
758758
self.skipTest("This test requires an xla_version >= 3.")
759-
if config.use_shardy_partitioner.value:
760-
self.skipTest("XLA failure due to b/370786664 and b/366411266. "
761-
"Enable when fixed.")
762759
mesh = jtu.create_mesh((4,), ('data'))
763760

764761
tpu_sharding = NamedSharding(mesh, P('data'))

0 commit comments

Comments
 (0)