x

docker/Dockerfile

@@ -192,14 +192,6 @@ RUN set -ex && \
     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 && \
     # cmake 配置
-    # ---- 诊断 HIP cmake（构建成功后删除）----
-    echo "=== HIP cmake files ===" && \
-    (find /opt/rocm/lib/cmake -iname "*hip*" 2>/dev/null || true) && \
-    echo "=== All cmake packages ===" && \
-    (ls /opt/rocm/lib/cmake/ || true) && \
-    echo "=== hipcc check ===" && \
-    (which hipcc && hipcc --version || true) && \
-    echo "=== HIP diagnostic end ===" && \
     mkdir -p /opt/vllm_build && \
     cmake -S /opt/vllm -B /opt/vllm_build -G Ninja \
         -DCMAKE_BUILD_TYPE=RelWithDebInfo \
@@ -209,6 +201,8 @@ RUN set -ex && \
         -DROCM_PATH=/opt/rocm \
         -DCMAKE_HIP_ARCHITECTURES=${ARCH} \
         -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
+        -DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
+        -DHIP_PATH=/opt/rocm \
         -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \
     # ninja 编译（-j8，32GB 内存；若 < 16GB 请改为 -j2）
     cd /opt/vllm_build && ninja -j8 && \