xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-03-16 13:57:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Build/CI] Fixing 'docker run' to re-enable AMD CI tests. (#4642)

Browse Source
This commit is contained in:
Alexei-V-Ivanov-AMD 2024-05-07 11:23:17 -05:00 committed by GitHub
parent 63575bc2e1
commit 478aed5827
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.buildkite/run-amd-test.sh
View File

@ -40,5 +40,5 @@ docker run \
-e HF_TOKEN \
--name ${container_name} \
${container_name} \
/bin/bash -c $(echo $1 | sed "s/^'//" | sed "s/'$//")
/bin/bash -c "${@}"

8
.buildkite/test-pipeline.yaml
View File

@ -48,7 +48,7 @@ steps:
- pytest -v -s test_pynccl.py
- label: Engine Test
mirror_hardwares: [amd]
#mirror_hardwares: [amd]
command: pytest -v -s engine tokenization test_sequence.py test_config.py test_logger.py
- label: Entrypoints Test
@ -73,13 +73,13 @@ steps:
parallelism: 4
- label: Models Test
mirror_hardwares: [amd]
#mirror_hardwares: [amd]
commands:
- bash ../.buildkite/download-images.sh
- pytest -v -s models --ignore=models/test_llava.py --ignore=models/test_mistral.py
- label: Llava Test
mirror_hardwares: [amd]
#mirror_hardwares: [amd]
commands:
- bash ../.buildkite/download-images.sh
- pytest -v -s models/test_llava.py
@ -101,7 +101,7 @@ steps:
command: pytest -v -s worker
- label: Speculative decoding tests
mirror_hardwares: [amd]
#mirror_hardwares: [amd]
command: pytest -v -s spec_decode
- label: LoRA Test %N

2
.buildkite/test-template.j2
View File

@ -26,7 +26,7 @@ steps:
- label: "AMD: {{ step.label }}"
agents:
queue: amd
command: bash .buildkite/run-amd-test.sh "'cd {{ (step.working_dir or default_working_dir) | safe }} && {{ step.command or (step.commands | join(' && ')) | safe }}'"
command: bash .buildkite/run-amd-test.sh "cd {{ (step.working_dir or default_working_dir) | safe }} ; {{ step.command or (step.commands | join(" ; ")) | safe }}"
env:
DOCKER_BUILDKIT: "1"
{% endif %}