Skip to content

Commit bde5039

Browse files
gmagogsfmIsotr0py
andauthored
[CI] Add compile/test_multimodal_compile.py to CI (#28151)
Signed-off-by: Yanan Cao <[email protected]> Co-authored-by: Isotr0py <[email protected]>
1 parent d72299d commit bde5039

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -460,6 +460,7 @@ steps:
460460
- tests/compile
461461
commands:
462462
- pytest -v -s compile/test_basic_correctness.py
463+
- pytest -v -s compile/test_multimodal_compile.py
463464
- pytest -v -s compile/piecewise/
464465

465466
- label: PyTorch Fullgraph Test # 22min

tests/compile/test_multimodal_compile.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,12 @@
44

55
from vllm.compilation.counter import compilation_counter
66
from vllm.config.compilation import CompilationMode
7+
from vllm.platforms import current_platform
78

89

910
# forked needed to workaround https://github.com/vllm-project/vllm/issues/21073
1011
@pytest.mark.forked
12+
@pytest.mark.skipif(not current_platform.is_cuda(), reason="Skip if not cuda")
1113
def test_qwen2_5_vl_compilation(vllm_runner, monkeypatch):
1214
"""Test that Qwen2.5-VL vision submodules are compiled.
1315

0 commit comments

Comments
 (0)