diff --git a/.github/workflows/pr-test-pd-router.yml b/.github/workflows/pr-test-pd-router.yml index 6fa5d726e..03e313985 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/cu128 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.7.post1 + python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.7.post2 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 20db2715f..a719846a2 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -196,7 +196,7 @@ fi # Python tools RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install \ datamodel_code_generator \ - mooncake-transfer-engine==0.3.7.post1 \ + mooncake-transfer-engine==0.3.7.post2 \ pre-commit \ pytest \ black \ diff --git a/docker/b300.Dockerfile b/docker/b300.Dockerfile index 868806df9..54ee1bec9 100644 --- a/docker/b300.Dockerfile +++ b/docker/b300.Dockerfile @@ -5,7 +5,7 @@ ARG BRANCH_TYPE=remote # Python tools RUN python3 -m pip install --no-cache-dir \ datamodel_code_generator \ - mooncake-transfer-engine==0.3.7.post1 \ + mooncake-transfer-engine==0.3.7.post2 \ pre-commit \ pytest \ black \ diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index de91105ec..925eb46b2 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -67,7 +67,7 @@ ARG LLVM_BRANCH="MainOpSelV2" ARG LLVM_COMMIT="6520ace8227ffe2728148d5f3b9872a870b0a560" ARG MOONCAKE_REPO="https://github.com/kvcache-ai/Mooncake.git" -ARG MOONCAKE_COMMIT="7c22adb0d2c1616e3cfefb2dc6bf01c23c3df768" +ARG MOONCAKE_COMMIT="b6a841dc78c707ec655a563453277d969fb8f38d" ARG TILELANG_REPO="https://github.com/HaiShaw/tilelang.git" ARG TILELANG_BRANCH="dsv32-mi35x" diff --git a/scripts/ci/ci_install_dependency.sh b/scripts/ci/ci_install_dependency.sh index 6b5ef055e..0d1c07a12 100755 --- a/scripts/ci/ci_install_dependency.sh +++ b/scripts/ci/ci_install_dependency.sh @@ -120,7 +120,7 @@ fi # Show current packages $PIP_CMD list -$PIP_CMD install mooncake-transfer-engine==0.3.7.post1 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX +$PIP_CMD install mooncake-transfer-engine==0.3.7.post2 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX if [ "$IS_BLACKWELL" != "1" ]; then # For lmms_evals evaluating MMMU