From ff00ef309dc619752c4c22028583e36a6f7e37a1 Mon Sep 17 00:00:00 2001 From: westers Date: Sat, 20 Dec 2025 10:21:34 -0600 Subject: [PATCH] Fix Dockerfile.rocm to set VLLM_TARGET_DEVICE=rocm MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Without this environment variable, the build defaults to VLLM_TARGET_DEVICE=cuda as defined in vllm/envs.py, causing cmake to configure with -DVLLM_TARGET_DEVICE=cuda instead of -DVLLM_TARGET_DEVICE=rocm. This fix ensures ROCm Docker builds use the correct target device. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 Signed-off-by: westers --- docker/Dockerfile.rocm | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 4c09808a14333..b7245e8067783 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -51,6 +51,8 @@ FROM fetch_vllm_${REMOTE_VLLM} AS fetch_vllm # ----------------------- # vLLM build stages FROM fetch_vllm AS build_vllm +# Set target device to ROCm for build +ENV VLLM_TARGET_DEVICE=rocm # Build vLLM RUN cd vllm \ && python3 -m pip install -r requirements/rocm.txt \