[CI] Clean DeepSeek V4 tests and installation scripts (#26609)

This commit is contained in:
Baizhou Zhang
2026-05-28 15:22:59 -07:00
committed by GitHub
parent f838adb7d4
commit 435c4ffb30
13 changed files with 66 additions and 198 deletions
-8
View File
@@ -548,20 +548,12 @@ RUN --mount=type=cache,target=/root/.cache/pip \
# the `nixl` import path) but unconditionally requires nixl-cu12, so we install
# it with --no-deps and pair it with the matching nixl-cu12 / nixl-cu13 binary
# to avoid shipping wrong-CUDA libs on cu13 images.
# The upstream flash-mla packages are required for running deepseek-v4 models
RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
python3 -m pip install nixl nixl-cu12 --no-deps ; \
python3 -m pip install cuda-python==12.9 ; \
cd /sgl-workspace && git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla \
&& cd flash-mla && git submodule update --init --recursive \
&& pip install --no-build-isolation -v . ; \
elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \
python3 -m pip install nixl nixl-cu13 --no-deps ; \
python3 -m pip install cuda-python==13.2.0 ; \
cd /sgl-workspace && git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla \
&& ln -s /usr/local/cuda/include/cccl/cuda /usr/local/cuda/include/cuda \
&& cd flash-mla && git submodule update --init --recursive \
&& pip install --no-build-isolation -v . ; \
fi
# Add yank script