x
This commit is contained in:
+1
-11
@@ -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)."
|
||||
|
||||
Reference in New Issue
Block a user