cjw/mineru-rocm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
chenjw28
2026-06-04 15:03:25 +08:00
parent 4553b6c073
commit 9944d8cc34
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+6 -2
View File
@@ -164,10 +164,14 @@ RUN set -ex && \
"setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \ "setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \
# 克隆 vllm main # 克隆 vllm main
cd /opt && git clone --depth 1 https://github.com/vllm-project/vllm.git && \ 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 && \ cd /opt/vllm && \
if [ -f csrc/mamba/mamba_ssm/selective_scan.h ]; then \
sed -i '109,121s/^/\/\/ /' csrc/mamba/mamba_ssm/selective_scan.h && \ sed -i '109,121s/^/\/\/ /' csrc/mamba/mamba_ssm/selective_scan.h && \
echo "vllm mamba operator+ patch applied." && \ echo "vllm mamba operator+ patch applied."; \
else \
echo "vllm mamba selective_scan.h not found (upstream removed), skipping patch."; \
fi && \
# cmake 别名兜底 # cmake 别名兜底
mkdir -p /opt/rocm/lib/cmake/hiprand && \ 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' \ 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' \