Unverified Commit cf5d27e3 authored by Zaili Wang's avatar Zaili Wang Committed by GitHub
Browse files

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

parent 9ec6031d
......@@ -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
......
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