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: ...@@ -9,6 +9,7 @@ on:
- "test/**" - "test/**"
- "sgl-kernel/**" - "sgl-kernel/**"
- ".github/workflows/pr-test-xeon.yml" - ".github/workflows/pr-test-xeon.yml"
- "docker/xeon.Dockerfile"
pull_request: pull_request:
branches: [ main ] branches: [ main ]
paths: paths:
...@@ -17,6 +18,7 @@ on: ...@@ -17,6 +18,7 @@ on:
- "test/**" - "test/**"
- "sgl-kernel/**" - "sgl-kernel/**"
- ".github/workflows/pr-test-xeon.yml" - ".github/workflows/pr-test-xeon.yml"
- "docker/xeon.Dockerfile"
types: [synchronize, labeled] types: [synchronize, labeled]
workflow_dispatch: workflow_dispatch:
......
...@@ -4,9 +4,9 @@ SHELL ["/bin/bash", "-c"] ...@@ -4,9 +4,9 @@ SHELL ["/bin/bash", "-c"]
ARG SGLANG_REPO=https://github.com/sgl-project/sglang.git ARG SGLANG_REPO=https://github.com/sgl-project/sglang.git
ARG VER_SGLANG=main ARG VER_SGLANG=main
ARG VER_TORCH=2.7.1 ARG VER_TORCH=2.9.0
ARG VER_TORCHVISION=0.22.1 ARG VER_TORCHVISION=0.24.0
ARG VER_TRITON=3.3.1 ARG VER_TRITON=3.5.0
RUN apt-get update && \ RUN apt-get update && \
apt-get full-upgrade -y && \ apt-get full-upgrade -y && \
......
...@@ -89,7 +89,7 @@ cp pyproject_cpu.toml pyproject.toml ...@@ -89,7 +89,7 @@ cp pyproject_cpu.toml pyproject.toml
pip install --upgrade pip setuptools pip install --upgrade pip setuptools
conda install -y libsqlite==3.48.0 gperftools tbb libnuma numactl conda install -y libsqlite==3.48.0 gperftools tbb libnuma numactl
pip install . 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 # Build the CPU backend kernels
cd ../sgl-kernel 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