Skip to content

Commit 17b4d85

Browse files
authored
[CI][TPU] Skip structured outputs+spec decode tests on TPU (#17510)
Signed-off-by: mgoin <[email protected]>
1 parent 1144a8e commit 17b4d85

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/v1/entrypoints/llm/test_struct_output_generate.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,9 @@ def test_structured_output(
8181
):
8282
monkeypatch.setenv("VLLM_USE_V1", "1")
8383

84+
if current_platform.is_tpu() and speculative_config:
85+
pytest.skip("TPU does not support speculative decoding")
86+
8487
# Don't use eager execution on TPUs because we want to test for no
8588
# recompilation at runtime
8689
enforce_eager = bool(not current_platform.is_tpu())

0 commit comments

Comments
 (0)