From c63814b3954a934aa43251de48b9883c6ac1a8f3 Mon Sep 17 00:00:00 2001
From: Kanglan Tang <kanglan@google.com>
Date: Fri, 7 Nov 2025 14:33:08 -0800
Subject: [PATCH] Skip `test_oom_compilation_only` on TPU v6e and v7x

PiperOrigin-RevId: 829579818
---
 tests/pallas/tpu_pallas_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/pallas/tpu_pallas_test.py b/tests/pallas/tpu_pallas_test.py
index 4e17d37b2fbb..344ba4bc9adf 100644
--- a/tests/pallas/tpu_pallas_test.py
+++ b/tests/pallas/tpu_pallas_test.py
@@ -2269,6 +2269,8 @@ def test_automatic_single_buffering(self,):
       self.skipTest('OOM tests need us to compile the kernels')
     if not jtu.if_cloud_tpu_at_least(2025, 11, 12):
       self.skipTest('Support added on Oct 14, 2025')
+    if jtu.is_device_tpu_at_least(6):
+      self.skipTest('Test currently fails on TPU v6e and v7x')
 
     def body(y_ref):
       pass  # We only want to compile the kernel.