Skip to content

Commit

Permalink
[ci] Add AMD, Neuron, Intel tests for AWS CI and turn off default sof…
Browse files Browse the repository at this point in the history
…t fail for GPU tests (vllm-project#5464)

Signed-off-by: kevin <kevin@anyscale.com>
  • Loading branch information
khluu authored Jun 12, 2024
1 parent c3c2903 commit 8b82a89
Showing 1 changed file with 29 additions and 1 deletion.
30 changes: 29 additions & 1 deletion .buildkite/test-template-aws.j2
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,34 @@ steps:
limit: 5
- wait

- group: "AMD Tests"
depends_on: ~
steps:
{% for step in steps %}
{% if step.mirror_hardwares and "amd" in step.mirror_hardwares %}
- 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 }}"
env:
DOCKER_BUILDKIT: "1"
soft_fail: true
{% endif %}
{% endfor %}

- label: "Neuron Test"
depends_on: ~
agents:
queue: neuron
command: bash .buildkite/run-neuron-test.sh
soft_fail: false

- label: "Intel Test"
depends_on: ~
agents:
queue: intel
command: bash .buildkite/run-cpu-test.sh

{% for step in steps %}
- label: "{{ step.label }}"
agents:
Expand All @@ -31,7 +59,7 @@ steps:
{% else %}
queue: gpu_1_queue
{% endif %}
soft_fail: true
soft_fail: {{ step.soft_fail or false }}
{% if step.parallelism %}
parallelism: {{ step.parallelism }}
{% endif %}
Expand Down

0 comments on commit 8b82a89

Please sign in to comment.