x

docker/Dockerfile

@@ -164,10 +164,14 @@ RUN set -ex && \
         "setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \
     # 克隆 vllm main
     cd /opt && git clone --depth 1 https://github.com/vllm-project/vllm.git && \
-    # 补丁 5：注释掉 vllm mamba 模块的 operator+ 定义
+    # 补丁 5：注释掉 vllm mamba 模块的 operator+ 定义（可能已被 vllm 移除）
     cd /opt/vllm && \
-    sed -i '109,121s/^/\/\/ /' csrc/mamba/mamba_ssm/selective_scan.h && \
-    echo "vllm mamba operator+ patch applied." && \
+    if [ -f csrc/mamba/mamba_ssm/selective_scan.h ]; then \
+        sed -i '109,121s/^/\/\/ /' csrc/mamba/mamba_ssm/selective_scan.h && \
+        echo "vllm mamba operator+ patch applied."; \
+    else \
+        echo "vllm mamba selective_scan.h not found (upstream removed), skipping patch."; \
+    fi && \
     # cmake 别名兜底
     mkdir -p /opt/rocm/lib/cmake/hiprand && \
     printf 'include(/opt/rocm/lib/cmake/rocrand/rocrand-config.cmake)\nif(TARGET roc::rocrand AND NOT TARGET hip::hiprand)\n  add_library(hip::hiprand ALIAS roc::rocrand)\nendif()\n' \