fix ci by skipping pallas-cuda on new tests

v0i0 · v0i0 · commit fc47a36b5e16 · 2026-03-24T20:03:29.000Z
diff --git a/test/inductor/test_pallas.py b/test/inductor/test_pallas.py
@@ -2164,6 +2164,7 @@ def transformer(x, mask, *layer_params):
         expected = transformer(x, mask, *all_params)
         self.assertEqual(result, expected, atol=atol, rtol=rtol)
 
+    @skip_if_cuda
     def test_transformer_tiny(self):
         """Test a 4-layer Llama-style transformer at tiny dimensions."""
         self._run_transformer(
@@ -2177,6 +2178,7 @@ def test_transformer_tiny(self):
             rtol=1e-2,
         )
 
+    @skip_if_cuda
     def test_transformer_medium(self):
         """Test a 4-layer transformer at Llama-7B-like dimensions."""
         self._run_transformer(
@@ -2454,6 +2456,7 @@ def fn(indices, table):
         expected = fn(indices, table)
         self.assertEqual(result, expected)
 
+    @skip_if_cuda
     def test_transformer_with_final_norm_and_lm_head(self):
         """Test multi-layer transformer + final RMSNorm + LM head (no embedding)."""
         torch._dynamo.reset()
