[compile] Add some more startup tests for top models (#38046)

Signed-off-by: Richard Zou <zou3519@gmail.com>
This commit is contained in:
Richard Zou
2026-03-25 12:02:22 -04:00
committed by GitHub
parent 1bf2ddd0ee
commit 6e37c46b35
4 changed files with 260 additions and 88 deletions

View File

@@ -17,6 +17,16 @@ steps:
# (using -0 for proper path handling)
- "find compile/ -maxdepth 1 -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
- label: PyTorch Compilation Unit Tests (H100)
timeout_in_minutes: 30
device: h100
num_devices: 1
source_file_dependencies:
- vllm/
- tests/compile/h100/
commands:
- "find compile/h100/ -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
- label: PyTorch Compilation Passes Unit Tests
timeout_in_minutes: 20
source_file_dependencies:
@@ -54,4 +64,4 @@ steps:
source_file_dependencies:
- requirements/nightly_torch_test.txt
commands:
- bash standalone_tests/pytorch_nightly_dependency.sh
- bash standalone_tests/pytorch_nightly_dependency.sh