@@ -250,10 +250,9 @@ steps:
250250 depends_on: block-amd-build
251251 soft_fail: true
252252 commands:
253- # Handle the introduction of test target in Dockerfile.rocm
254253 - "grep -i 'from base as test' Dockerfile.rocm && docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --target test --progress plain . || docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --progress plain ."
255254 - "docker push {{ docker_image_amd }}"
256- key: " amd-build"
255+ key: amd-build
257256 env:
258257 DOCKER_BUILDKIT: "1"
259258 retry:
@@ -266,43 +265,25 @@ steps:
266265 limit: 1
267266 agents:
268267 queue: amd-cpu
269-
270- {% for step in steps %}
271- {% if step .mirror_hardwares and "amd" in step .mirror_hardwares %}
272- {% if step .label == "placeholder" %}
273- - label: "AMD: {{ step.label }}"
274- depends_on: amd-build
275- agents:
276- {% if step .label and step .label =="Benchmarks" %}
277- queue: amd_mi300
278- {% else %}
279- queue: amd_gpu
280- {% endif %}
281-
282- command: bash .buildkite/run-amd-test.sh "(command rocm-smi || true) && export VLLM_LOGGING_LEVEL=DEBUG && export VLLM_ALLOW_DEPRECATED_BEAM_SEARCH=1 && cd {{ (step.working_dir or default_working_dir) | safe }} ; {{ step.command or (step.commands | join(" && ")) | safe }}"
283- env:
284- DOCKER_BUILDKIT: "1"
285- priority: 100
286- soft_fail: true
287- {% else %}
288- - block: "Run AMD: {{ step.label }}"
289- depends_on: amd-build
290- key: block-amd-{{ step.label | replace(" ", "-") | lower | replace("(", "") | replace(")", "") | replace("%", "") | replace(",", "-") }}
291-
292- - label: "AMD: {{ step.label }}"
293- depends_on: block-amd-{{ step.label | replace(" ", "-") | lower | replace("(", "") | replace(")", "") | replace("%", "") | replace(",", "-") }}
294- agents:
295- {% if step .label and step .label =="Benchmarks" %}
296- queue: amd_mi300
297- {% else %}
298- queue: amd_gpu
299- {% endif %}
300-
301- command: bash .buildkite/run-amd-test.sh "(command rocm-smi || true) && export VLLM_LOGGING_LEVEL=DEBUG && export VLLM_ALLOW_DEPRECATED_BEAM_SEARCH=1 && cd {{ (step.working_dir or default_working_dir) | safe }} ; {{ step.command or (step.commands | join(" && ")) | safe }}"
302- env:
303- DOCKER_BUILDKIT: "1"
304- priority: 100
305- soft_fail: true
306- {% endif %}
307- {% endif %}
308- {% endfor %}
268+ {% for step in steps %}
269+ {% if step .mirror_hardwares and "amd" in step .mirror_hardwares %}
270+ - block: "Run AMD: {{ step.label }}"
271+ depends_on: amd-build
272+ key: block-amd-{{ step.label | replace(" ", "-") | lower | replace("(", "") | replace(")", "") | replace("%", "") | replace(",", "-") }}
273+
274+ - label: "AMD: {{ step.label }}"
275+ depends_on: block-amd-{{ step.label | replace(" ", "-") | lower | replace("(", "") | replace(")", "") | replace("%", "") | replace(",", "-") }}
276+ agents:
277+ {% if step .label and step .label =="Benchmarks" %}
278+ queue: amd_mi300
279+ {% else %}
280+ queue: amd_gpu
281+ {% endif %}
282+
283+ command: bash .buildkite/run-amd-test.sh "(command rocm-smi || true) && export VLLM_LOGGING_LEVEL=DEBUG && export VLLM_ALLOW_DEPRECATED_BEAM_SEARCH=1 && cd {{ (step.working_dir or default_working_dir) | safe }} ; {{ step.command or (step.commands | join(" && ")) | safe }}"
284+ env:
285+ DOCKER_BUILDKIT: "1"
286+ priority: 100
287+ soft_fail: true
288+ {% endif %}
289+ {% endfor %}
0 commit comments