[ci] Switch some CI jobs to H200 MIG slices (#38956)

This commit is contained in:
Kevin H. Luu
2026-04-05 13:26:11 -07:00
committed by GitHub
parent 4dd49b06f8
commit 56de443db1
14 changed files with 25 additions and 0 deletions

View File

@@ -49,6 +49,7 @@ steps:
- label: PyTorch Fullgraph
timeout_in_minutes: 30
device: h200_18gb
source_file_dependencies:
- vllm/
- tests/compile
@@ -60,6 +61,7 @@ steps:
# if this test fails, it means the nightly torch version is not compatible with some
# of the dependencies. Please check the error message and add the package to whitelist
# in /vllm/tools/pre_commit/generate_nightly_torch_test.py
device: h200_18gb
soft_fail: true
source_file_dependencies:
- requirements/nightly_torch_test.txt