fix: add CMAKE_LIBRARY_PATH=/opt/rocm/lib for -lMIOpen
This commit is contained in:
+2
-1
@@ -253,7 +253,8 @@ 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" && \
|
||||
-DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" \
|
||||
-DCMAKE_LIBRARY_PATH=/opt/rocm/lib && \
|
||||
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