[Build/CI] Fixing 'docker run' to re-enable AMD CI tests. (#4642)
This commit is contained in:
committed by
GitHub
parent
63575bc2e1
commit
478aed5827
@@ -40,5 +40,5 @@ docker run \
|
|||||||
-e HF_TOKEN \
|
-e HF_TOKEN \
|
||||||
--name ${container_name} \
|
--name ${container_name} \
|
||||||
${container_name} \
|
${container_name} \
|
||||||
/bin/bash -c $(echo $1 | sed "s/^'//" | sed "s/'$//")
|
/bin/bash -c "${@}"
|
||||||
|
|
||||||
|
|||||||
@@ -48,7 +48,7 @@ steps:
|
|||||||
- pytest -v -s test_pynccl.py
|
- pytest -v -s test_pynccl.py
|
||||||
|
|
||||||
- label: Engine Test
|
- label: Engine Test
|
||||||
mirror_hardwares: [amd]
|
#mirror_hardwares: [amd]
|
||||||
command: pytest -v -s engine tokenization test_sequence.py test_config.py test_logger.py
|
command: pytest -v -s engine tokenization test_sequence.py test_config.py test_logger.py
|
||||||
|
|
||||||
- label: Entrypoints Test
|
- label: Entrypoints Test
|
||||||
@@ -73,13 +73,13 @@ steps:
|
|||||||
parallelism: 4
|
parallelism: 4
|
||||||
|
|
||||||
- label: Models Test
|
- label: Models Test
|
||||||
mirror_hardwares: [amd]
|
#mirror_hardwares: [amd]
|
||||||
commands:
|
commands:
|
||||||
- bash ../.buildkite/download-images.sh
|
- bash ../.buildkite/download-images.sh
|
||||||
- pytest -v -s models --ignore=models/test_llava.py --ignore=models/test_mistral.py
|
- pytest -v -s models --ignore=models/test_llava.py --ignore=models/test_mistral.py
|
||||||
|
|
||||||
- label: Llava Test
|
- label: Llava Test
|
||||||
mirror_hardwares: [amd]
|
#mirror_hardwares: [amd]
|
||||||
commands:
|
commands:
|
||||||
- bash ../.buildkite/download-images.sh
|
- bash ../.buildkite/download-images.sh
|
||||||
- pytest -v -s models/test_llava.py
|
- pytest -v -s models/test_llava.py
|
||||||
@@ -101,7 +101,7 @@ steps:
|
|||||||
command: pytest -v -s worker
|
command: pytest -v -s worker
|
||||||
|
|
||||||
- label: Speculative decoding tests
|
- label: Speculative decoding tests
|
||||||
mirror_hardwares: [amd]
|
#mirror_hardwares: [amd]
|
||||||
command: pytest -v -s spec_decode
|
command: pytest -v -s spec_decode
|
||||||
|
|
||||||
- label: LoRA Test %N
|
- label: LoRA Test %N
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ steps:
|
|||||||
- label: "AMD: {{ step.label }}"
|
- label: "AMD: {{ step.label }}"
|
||||||
agents:
|
agents:
|
||||||
queue: amd
|
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:
|
env:
|
||||||
DOCKER_BUILDKIT: "1"
|
DOCKER_BUILDKIT: "1"
|
||||||
{% endif %}
|
{% endif %}
|
||||||
|
|||||||
Reference in New Issue
Block a user