[CPU] Upgrade default PT version to 2.9 (#12611)

This commit is contained in:
Zaili Wang
2025-11-06 10:16:35 +08:00
committed by GitHub
parent 9ec6031d70
commit cf5d27e30a
3 changed files with 6 additions and 4 deletions

View File

@@ -9,6 +9,7 @@ on:
- "test/**"
- "sgl-kernel/**"
- ".github/workflows/pr-test-xeon.yml"
- "docker/xeon.Dockerfile"
pull_request:
branches: [ main ]
paths:
@@ -17,6 +18,7 @@ on:
- "test/**"
- "sgl-kernel/**"
- ".github/workflows/pr-test-xeon.yml"
- "docker/xeon.Dockerfile"
types: [synchronize, labeled]
workflow_dispatch:

View File

@@ -4,9 +4,9 @@ SHELL ["/bin/bash", "-c"]
ARG SGLANG_REPO=https://github.com/sgl-project/sglang.git
ARG VER_SGLANG=main
ARG VER_TORCH=2.7.1
ARG VER_TORCHVISION=0.22.1
ARG VER_TRITON=3.3.1
ARG VER_TORCH=2.9.0
ARG VER_TORCHVISION=0.24.0
ARG VER_TRITON=3.5.0
RUN apt-get update && \
apt-get full-upgrade -y && \

View File

@@ -89,7 +89,7 @@ cp pyproject_cpu.toml pyproject.toml
pip install --upgrade pip setuptools
conda install -y libsqlite==3.48.0 gperftools tbb libnuma numactl
pip install .
pip install torch==2.7.1 torchvision==0.22.1 triton==3.3.1 --force-reinstall
pip install torch==2.9.0 torchvision==0.24.0 triton==3.5.0 --force-reinstall
# Build the CPU backend kernels
cd ../sgl-kernel