diff --git a/docker/Dockerfile b/docker/Dockerfile index 4e1518c..732065e 100644 --- a/docker/Dockerfile +++ b/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' \