We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c098b36 commit 8bbd738Copy full SHA for 8bbd738
tests/memories_test.py
@@ -756,9 +756,6 @@ def init():
756
def test_compute_no_inputs_host_replicated(self):
757
if xb.backend_xla_version() is not None and xb.backend_xla_version() < 3:
758
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.")
762
mesh = jtu.create_mesh((4,), ('data'))
763
764
tpu_sharding = NamedSharding(mesh, P('data'))
0 commit comments