| 1 | group: Torch ABI |
| 2 | depends_on: |
| 3 | - image-build |
| 4 | steps: |
| 5 | - label: ":nvidia: (L4) Torch Stable ABI Audit" |
| 6 | key: torch-stable-abi-audit |
| 7 | timeout_in_minutes: 10 |
| 8 | source_file_dependencies: |
| 9 | - .buildkite/check-torch-abi.py |
| 10 | - csrc/ |
| 11 | - cmake/ |
| 12 | - setup.py |
| 13 | commands: |
| 14 | - python3 /vllm-workspace/.buildkite/check-torch-abi.py |
| 15 | mirror: |
| 16 | amd: |
| 17 | label: ":amd: (MI250) Torch Stable ABI Audit" |
| 18 | dind: false |
| 19 | device: mi300_1 |
| 20 | timeout_in_minutes: 35 |
| 21 | depends_on: |
| 22 | - image-build-amd |
| 23 | no_gpu: true |
| 24 | working_dir: /vllm-workspace/ |
| 25 | source_file_dependencies: |
| 26 | - .buildkite/check-torch-abi.py |
| 27 | - csrc/ |
| 28 | - cmake/ |
| 29 | - setup.py |
| 30 | - vllm/platforms/rocm.py |
| 31 | commands: |
| 32 | - uv pip install --system torch-abi-audit==0.0.1 |
| 33 | - python3 /vllm-workspace/.buildkite/check-torch-abi.py |