Unverified Commit 478aed58 authored by Alexei-V-Ivanov-AMD's avatar Alexei-V-Ivanov-AMD Committed by GitHub
Browse files

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

parent 63575bc2
......@@ -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 "${@}"
......@@ -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
......
......@@ -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 %}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment