@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / 06fd9ffcc4
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 5, 13:30 UTC

[ROCm][CI] Fix ROCm Dockerfile conftest generation for older Docker parsers (#38959)

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

Andreas Karatzas committed Apr 3, 2026 at 21:41 UTC 06fd9ffcc4ce51f38a482ca406abcf1d42cc2757
1 file changed +15 -15
docker/Dockerfile.rocm
+15 -15
@@ -390,21 +390,21 @@ ENV MIOPEN_DEBUG_CONV_GEMM=0
390 RUN mkdir src && mv vllm src/vllm
391
392 # This is a workaround to ensure pytest exits with the correct status code in CI tests.
393 -RUN cat << 'EOF' > /vllm-workspace/conftest.py
394 -import os
395 -
396 -_exit_code = 1
397 -
398 -def pytest_sessionfinish(session, exitstatus):
399 - global _exit_code
400 - _exit_code = int(exitstatus)
401 -
402 -def pytest_unconfigure(config):
403 - import sys
404 - sys.stdout.flush()
405 - sys.stderr.flush()
406 - os._exit(_exit_code)
407 -EOF
393 +RUN printf '%s\n' \
394 + 'import os' \
395 + '' \
396 + '_exit_code = 1' \
397 + '' \
398 + 'def pytest_sessionfinish(session, exitstatus):' \
399 + ' global _exit_code' \
400 + ' _exit_code = int(exitstatus)' \
401 + '' \
402 + 'def pytest_unconfigure(config):' \
403 + ' import sys' \
404 + ' sys.stdout.flush()' \
405 + ' sys.stderr.flush()' \
406 + ' os._exit(_exit_code)' \
407 + > /vllm-workspace/conftest.py
408
409 # -----------------------
410 # Final vLLM image