diff --git a/.github/workflows/pr-test-xeon.yml b/.github/workflows/pr-test-xeon.yml index f3a6cae7c..c2f1a9b85 100644 --- a/.github/workflows/pr-test-xeon.yml +++ b/.github/workflows/pr-test-xeon.yml @@ -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: diff --git a/docker/xeon.Dockerfile b/docker/xeon.Dockerfile index 17572397d..c0d82ffb9 100644 --- a/docker/xeon.Dockerfile +++ b/docker/xeon.Dockerfile @@ -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 && \ diff --git a/docs/platforms/cpu_server.md b/docs/platforms/cpu_server.md index 6a04897e1..5b86c8288 100644 --- a/docs/platforms/cpu_server.md +++ b/docs/platforms/cpu_server.md @@ -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