diff --git a/docker/Dockerfile b/docker/Dockerfile index ee3bf5a..b0e53b8 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -184,6 +184,7 @@ RUN set -ex && \ -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ -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" && \ cd /opt/vllm_build && ninja -j8 && \ cp /opt/vllm_build/*.abi3.so /opt/vllm/vllm/ && \