[ROCm][CI] Prepare TheRock image for CI (#56921)
Signed-off-by: Matthew Wong <Matthew.Wong2@amd.com> Co-authored-by: Andreas Karatzas <akaratza@amd.com>
Matt committed
Sep 14, 2026 at 18:57 UTC
31928987546ff6454f9a7d945f3ff132c5f9c531
2 files changed
+18
docker/Dockerfile.rock
+9
@@ -875,6 +875,15 @@ RUN mkdir src && mv vllm src/vllm
875
# Catch GPU<->CPU syncs in execute_model/sample_tokens during tests.
876
ENV VLLM_GPU_SYNC_CHECK=error
877
878
+FROM test AS test_smoke
879
+ARG ROCM_SMOKE_ID
880
+RUN --network=none \
881
+ bash /vllm-workspace/.buildkite/scripts/rocm/smoke-test-image.sh --inside \
882
+ && printf '%s\n' "${ROCM_SMOKE_ID:-local}" > /vllm-smoke-ok
883
+
884
+FROM scratch AS export_test_smoke
885
+COPY --from=test_smoke /vllm-smoke-ok /
886
+
887
# -----------------------
888
# Final vLLM image
889
FROM mori_base AS final_common
docker/Dockerfile.rock_base
+9
@@ -146,6 +146,15 @@ RUN if [ -f "${SITE_PACKAGES}/rocm_sdk/__init__.py" ]; then \
146
ln -sfn "${ROCM_PATH}/lib/${bn}" "$f"; \
147
fi; \
148
done \
149
+ && for f in librocprofiler-sdk.so.1 librocprofiler-register.so.0 \
150
+ librocprofiler-sdk-roctx.so.1; do \
151
+ if [ -e "${SITE_PACKAGES}/_rocm_sdk_core/lib/${f}" ] \
152
+ && [ -e "${SITE_PACKAGES}/_rocm_sdk_devel/lib/${f}" ]; then \
153
+ rm -f "${SITE_PACKAGES}/_rocm_sdk_core/lib/${f}"; \
154
+ ln -sfn "${SITE_PACKAGES}/_rocm_sdk_devel/lib/${f}" \
155
+ "${SITE_PACKAGES}/_rocm_sdk_core/lib/${f}"; \
156
+ fi; \
157
+ done \
158
&& ldconfig
159
160
# Install sccache if USE_SCCACHE is enabled (for release builds)