Skip to content

Commit 1a78757

Browse files
apaszkeGoogle-ML-Automation
authored andcommitted
Add a missing libtpu version skip in SC Pallas tests
PiperOrigin-RevId: 842744528
1 parent 11bb981 commit 1a78757

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/pallas/tpu_sparsecore_pallas_test.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1925,6 +1925,11 @@ class PipelineTestWithTCTiling(TCTilingMixin, PipelineTest):
19251925

19261926
class PallasSparsecoreAsyncTest(PallasSCTest):
19271927

1928+
def setUp(self):
1929+
super().setUp()
1930+
if not jtu.is_cloud_tpu_at_least(2025, 12, 14):
1931+
self.skipTest("Needs a newer libtpu")
1932+
19281933
@parameterized.product(
19291934
shape=[
19301935
(8, 128),

0 commit comments

Comments
 (0)