beef814e1d
Signed-off-by: Andreas Karatzas <akaratza@amd.com> Signed-off-by: Stefan Koncarevic <Stefan.Koncarevic@amd.com> Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com> Co-authored-by: Stefan Koncarevic <Stefan.Koncarevic@amd.com> Co-authored-by: OpenAI Codex <codex@openai.com> Co-authored-by: OpenAI Codex <noreply@openai.com>
34 lines
838 B
YAML
34 lines
838 B
YAML
group: Torch ABI
|
|
depends_on:
|
|
- image-build
|
|
steps:
|
|
- label: ":nvidia: (L4) Torch Stable ABI Audit"
|
|
key: torch-stable-abi-audit
|
|
timeout_in_minutes: 10
|
|
source_file_dependencies:
|
|
- .buildkite/check-torch-abi.py
|
|
- csrc/
|
|
- cmake/
|
|
- setup.py
|
|
commands:
|
|
- python3 /vllm-workspace/.buildkite/check-torch-abi.py
|
|
mirror:
|
|
amd:
|
|
label: ":amd: (MI250) Torch Stable ABI Audit"
|
|
dind: false
|
|
device: mi300_1
|
|
timeout_in_minutes: 35
|
|
depends_on:
|
|
- image-build-amd
|
|
no_gpu: true
|
|
working_dir: /vllm-workspace/
|
|
source_file_dependencies:
|
|
- .buildkite/check-torch-abi.py
|
|
- csrc/
|
|
- cmake/
|
|
- setup.py
|
|
- vllm/platforms/rocm.py
|
|
commands:
|
|
- uv pip install --system torch-abi-audit==0.0.1
|
|
- python3 /vllm-workspace/.buildkite/check-torch-abi.py
|