@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / 4fccd30f19
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 6, 19:15 UTC

[ROCm][CI] Upgrading orchestrator to handle python pipeline markers and options (#36181)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>

Andreas Karatzas committed Mar 13, 2026 at 04:04 UTC 4fccd30f19e0b44ec4a2b076cfc33aeafdd2d72e
1 file changed +14
.buildkite/scripts/hardware_ci/run-amd-test.sh
+14
@@ -205,6 +205,13 @@ re_quote_pytest_markers() {
205 esac
206
207 if $is_boundary; then
208 + # Strip surrounding double quotes if present (from upstream
209 + # single-to-double conversion); without this, wrapping below
210 + # would produce '"expr"' with literal double-quote characters.
211 + if [[ "$marker_buf" == '"'*'"' ]]; then
212 + marker_buf="${marker_buf#\"}"
213 + marker_buf="${marker_buf%\"}"
214 + fi
215 # Flush the collected marker expression
216 if [[ "$marker_buf" == *" "* || "$marker_buf" == *"("* ]]; then
217 output+="'${marker_buf}' "
@@ -242,6 +249,11 @@ re_quote_pytest_markers() {
249
250 # Flush any trailing marker expression (marker at end of command)
251 if $collecting && [[ -n "$marker_buf" ]]; then
252 + # Strip surrounding double quotes (see mid-stream flush comment)
253 + if [[ "$marker_buf" == '"'*'"' ]]; then
254 + marker_buf="${marker_buf#\"}"
255 + marker_buf="${marker_buf%\"}"
256 + fi
257 if [[ "$marker_buf" == *" "* || "$marker_buf" == *"("* ]]; then
258 output+="'${marker_buf}'"
259 else
@@ -492,6 +504,8 @@ else
504 -e HF_TOKEN \
505 -e AWS_ACCESS_KEY_ID \
506 -e AWS_SECRET_ACCESS_KEY \
507 + -e BUILDKITE_PARALLEL_JOB \
508 + -e BUILDKITE_PARALLEL_JOB_COUNT \
509 -v "${HF_CACHE}:${HF_MOUNT}" \
510 -e "HF_HOME=${HF_MOUNT}" \
511 -e "PYTHONPATH=${MYPYTHONPATH}" \