x

2026-06-05 16:15:17 +08:00
parent 5fc53646b2
commit 95e7ac8f0f
1 changed files with 5 additions and 0 deletions
@@ -161,7 +161,12 @@ RUN set -ex && \
    mkdir -p /opt/rocm/lib/cmake/hipblas && \
    printf 'include(/opt/rocm/lib/cmake/rocblas/rocblas-config.cmake)\nif(TARGET roc::rocblas AND NOT TARGET hip::hipblas)\n  add_library(hip::hipblas ALIAS roc::rocblas)\nendif()\n' \
        > /opt/rocm/lib/cmake/hipblas/hipblas-config.cmake && \
+    # 创建 HIP cmake config（apt 安装的 hip-dev 缺少 find_package 需要的配置）
+    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 && \
    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 \
        -DCMAKE_BUILD_TYPE=RelWithDebInfo \
        -DVLLM_TARGET_DEVICE=rocm \