This commit is contained in:
2026-06-09 11:22:04 +08:00
parent cccfa8830a
commit 9c38cdad28
2 changed files with 99 additions and 13 deletions
+1 -11
View File
@@ -242,7 +242,6 @@ RUN set -ex && \
fi && \
mkdir -p /opt/vllm_build && \
export HIP_PLATFORM=amd HIP_PATH=/opt/rocm ROCM_PATH=/opt/rocm && \
CMAKE_EC=0; \
cmake -S /opt/vllm -B /opt/vllm_build -G Ninja \
-DCMAKE_BUILD_TYPE=RelWithDebInfo \
-DVLLM_TARGET_DEVICE=rocm \
@@ -254,16 +253,7 @@ RUN set -ex && \
-DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
-DHIP_PATH=/opt/rocm \
-DHIP_INCLUDE_DIR=/opt/rocm/include \
-DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" \
>/tmp/cmake.log 2>&1 || CMAKE_EC=$?; \
cat /tmp/cmake.log; \
if [ $CMAKE_EC -ne 0 ]; then \
echo ""; \
echo "========== CMake FAILED (exit=$CMAKE_EC) =========="; \
echo "=== grep CMake Error / fatal ==="; \
grep -i 'cmake error\|CMake Error\|fatal' /tmp/cmake.log || true; \
exit $CMAKE_EC; \
fi && \
-DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \
cd /opt/vllm_build && ninja -j8 && \
cp /opt/vllm_build/*.abi3.so /opt/vllm/vllm/ && \
echo "vllm C++ build complete (layer cached)."