[CPU] Upgrade default PT version to 2.9 (#12611)
This commit is contained in:
@@ -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:
|
|||||||
- "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"]
|
|||||||
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
|
|||||||
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
|
||||||
|
|||||||
Reference in New Issue
Block a user