Unverified Commit ab8b83f7 authored by Shangming Cai's avatar Shangming Cai Committed by GitHub
Browse files

chore: upgrade mooncake 0.3.7.post1 (#12541)


Signed-off-by: default avatarShangming Cai <csmthu@gmail.com>
parent de0b10cf
......@@ -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.6.post1
python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.7.post1
python3 -m pip --no-cache-dir install --user --force-reinstall genai-bench==0.0.2
- name: Build and install sgl-router
......
......@@ -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.6.post1 \
mooncake-transfer-engine==0.3.7.post1 \
pre-commit \
pytest \
black \
......
......@@ -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.6.post1 \
mooncake-transfer-engine==0.3.7.post1 \
pre-commit \
pytest \
black \
......
......@@ -10,10 +10,6 @@ export LD_LIBRARY_PATH="${NVSHMEM_DIR}/lib:$LD_LIBRARY_PATH"
export PATH="${NVSHMEM_DIR}/bin:$PATH"
export CUDA_HOME=/usr/local/cuda
# Install Mooncake+EP
curl -L https://cloud.tsinghua.edu.cn/f/c22ec766545e48bf99e8/?dl=1 -o mooncake_transfer_engine-0.3.6.post1+ep-cp310-cp310-manylinux_2_17_x86_64.manylinux_2_35_x86_64.whl
UV_SYSTEM_PYTHON=true uv pip install mooncake_transfer_engine-0.3.6.post1+ep-cp310-cp310-manylinux_2_17_x86_64.manylinux_2_35_x86_64.whl
if python3 -c "import deep_ep" >/dev/null 2>&1; then
echo "deep_ep is already installed or importable. Skipping installation."
exit 0
......
......@@ -120,7 +120,7 @@ fi
# Show current packages
$PIP_CMD list
$PIP_CMD install mooncake-transfer-engine==0.3.6.post1 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX
$PIP_CMD install mooncake-transfer-engine==0.3.7.post1 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX
if [ "$IS_BLACKWELL" != "1" ]; then
# For lmms_evals evaluating MMMU
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment