From 74cd6e390259546913a466a9dbfa53e651c6e20b Mon Sep 17 00:00:00 2001 From: Shangming Cai Date: Sat, 20 Sep 2025 15:12:26 +0800 Subject: [PATCH] chore: upgrade mooncake 0.3.6.post1 to fix gb200 dockerfile (#10681) Signed-off-by: Shangming Cai --- .github/workflows/pr-test-pd-router.yml | 2 +- docker/Dockerfile | 2 +- docker/Dockerfile.gb200 | 2 +- scripts/ci/ci_install_dependency.sh | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/pr-test-pd-router.yml b/.github/workflows/pr-test-pd-router.yml index 300b391a7..da4babeef 100644 --- a/.github/workflows/pr-test-pd-router.yml +++ b/.github/workflows/pr-test-pd-router.yml @@ -140,7 +140,7 @@ jobs: python3 -m pip --no-cache-dir install --upgrade pip python3 -m pip --no-cache-dir install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu126 python3 -m pip --no-cache-dir install -e "python[all]" --break-system-packages - python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.6 + python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.6.post1 python3 -m pip --no-cache-dir install --user --force-reinstall genai-bench==0.0.2 - name: Build and install sgl-router diff --git a/docker/Dockerfile b/docker/Dockerfile index 40b9edbb8..2f4c22d90 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -128,7 +128,7 @@ RUN cd /sgl-workspace/DeepEP && \ # Python tools RUN python3 -m pip install --no-cache-dir \ datamodel_code_generator \ - mooncake-transfer-engine==0.3.6 \ + mooncake-transfer-engine==0.3.6.post1 \ pre-commit \ pytest \ black \ diff --git a/docker/Dockerfile.gb200 b/docker/Dockerfile.gb200 index cc0e9d92f..db9fa5be7 100644 --- a/docker/Dockerfile.gb200 +++ b/docker/Dockerfile.gb200 @@ -118,7 +118,7 @@ RUN cd /sgl-workspace/DeepEP && \ # Python tools RUN python3 -m pip install --no-cache-dir \ datamodel_code_generator \ - mooncake-transfer-engine==0.3.6 \ + mooncake-transfer-engine==0.3.6.post1 \ pre-commit \ pytest \ black \ diff --git a/scripts/ci/ci_install_dependency.sh b/scripts/ci/ci_install_dependency.sh index 0e370eec3..abc0b67cc 100755 --- a/scripts/ci/ci_install_dependency.sh +++ b/scripts/ci/ci_install_dependency.sh @@ -74,7 +74,7 @@ fi $PIP_CMD list # Install additional dependencies -$PIP_CMD install mooncake-transfer-engine==0.3.6 nvidia-cuda-nvrtc-cu12 py-spy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX +$PIP_CMD install mooncake-transfer-engine==0.3.6.post1 nvidia-cuda-nvrtc-cu12 py-spy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX if [ "$IS_BLACKWELL" != "1" ]; then # For lmms_evals evaluating MMMU