x

docker/Dockerfile

@@ -165,6 +165,13 @@ RUN set -ex && \
     mkdir -p /opt/rocm/lib/cmake/hip && \
     printf 'set(HIP_FOUND TRUE)\nset(HIP_VERSION "7.2.1")\nset(HIP_PLATFORM amd)\nset(HIP_COMPILER /opt/rocm/llvm/bin/clang++)\nset(HIP_RUNTIME rocm)\nset(HIP_INCLUDE_DIRS /opt/rocm/include)\nset(HIP_LIBRARIES /opt/rocm/lib/libamdhip64.so)\nif(NOT TARGET hip::host)\n  add_library(hip::host INTERFACE IMPORTED)\n  set_target_properties(hip::host PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${HIP_INCLUDE_DIRS}")\nendif()\nif(NOT TARGET hip::device)\n  add_library(hip::device INTERFACE IMPORTED)\n  set_target_properties(hip::device PROPERTIES INTERFACE_COMPILE_OPTIONS "${HIP_COMPILER}")\nendif()\n' \
         > /opt/rocm/lib/cmake/hip/hip-config.cmake && \
+    # 补丁: ROCm 7.2.1 中 hip_version.h 已移除, 创建兼容头文件供 PyTorch LoadHIP.cmake 使用
+    if [ ! -f /opt/rocm/include/hip/hip_version.h ]; then \
+        mkdir -p /opt/rocm/include/hip && \
+        printf '#pragma once\n#define HIP_VERSION_MAJOR 7\n#define HIP_VERSION_MINOR 2\n#define HIP_VERSION_PATCH 53211\n#define HIP_VERSION_GITDATE 0\n#define HIP_VERSION (HIP_VERSION_MAJOR * 10000000 + HIP_VERSION_MINOR * 100000 + HIP_VERSION_PATCH)\n' \
+            > /opt/rocm/include/hip/hip_version.h && \
+        echo "hip_version.h created."; \
+    fi && \
     mkdir -p /opt/vllm_build && \
     export HIP_PLATFORM=amd HIP_PATH=/opt/rocm ROCM_PATH=/opt/rocm && \
     cmake -S /opt/vllm -B /opt/vllm_build -G Ninja \
@@ -177,7 +184,6 @@ RUN set -ex && \
         -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
         -DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
         -DHIP_PATH=/opt/rocm \
-        -DCMAKE_INCLUDE_PATH=/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/ && \