Skip to content

Commit 37291bc

Browse files
danielsuoGoogle-ML-Automation
authored andcommitted
[pallas:tpu] Skip flaky tpu sparsecore test.
PiperOrigin-RevId: 833347091
1 parent f7ce3be commit 37291bc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/pallas/tpu_sparsecore_pallas_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -287,6 +287,7 @@ def kernel(x_ref, o_ref):
287287
np.testing.assert_array_equal(kernel(x), x.sum(axis=0))
288288

289289
def test_get_multi_index(self):
290+
self.skipTest("TODO(apaszke): This test may be flaky, at least for TPU v6.")
290291
self.skip_if_tc_tiling()
291292

292293
@self.vector_subcore_kernel(

0 commit comments

Comments
 (0)