[CPU] Upgrade default PT version to 2.9 (#12611)
This commit is contained in:
2
.github/workflows/pr-test-xeon.yml
vendored
2
.github/workflows/pr-test-xeon.yml
vendored
@@ -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:
|
||||
|
||||
|
||||
@@ -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 && \
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user