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 \ ...@@ -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: ...@@ -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: ...@@ -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: ...@@ -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: ...@@ -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 %}
......
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