minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Intel XPU]Update pytorch xpu to 2.9 (#12363)

Browse Source
This commit is contained in:
gaopengff
2025-11-06 10:34:23 -08:00
committed by GitHub
parent 4cc725ac1c
commit 4e234b4cf9
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/xpu.Dockerfile
+2 -2
View File
@@ -3,7 +3,7 @@
# Usage: docker build --build-arg UBUNTU_VERSION=24.04 --build-arg PYTHON_VERSION=3.10 -t sglang:xpu_kernel -f xpu.Dockerfile --no-cache . # Usage: docker build --build-arg UBUNTU_VERSION=24.04 --build-arg PYTHON_VERSION=3.10 -t sglang:xpu_kernel -f xpu.Dockerfile --no-cache .
# Use Intel deep learning essentials base image with Ubuntu 24.04 # Use Intel deep learning essentials base image with Ubuntu 24.04
FROM intel/deep-learning-essentials:2025.1.3-0-devel-ubuntu24.04 FROM intel/deep-learning-essentials:2025.2.2-0-devel-ubuntu24.04
# Avoid interactive prompts during package install # Avoid interactive prompts during package install
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
@@ -48,7 +48,7 @@ RUN --mount=type=secret,id=github_token \
cd /home/sdp && \ cd /home/sdp && \
. /home/sdp/miniforge3/bin/activate && \ . /home/sdp/miniforge3/bin/activate && \
conda activate py${PYTHON_VERSION} && \ conda activate py${PYTHON_VERSION} && \
pip3 install torch==2.8.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.4.0 --index-url https://download.pytorch.org/whl/xpu pip3 install torch==2.9.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.5.0 --index-url https://download.pytorch.org/whl/xpu
RUN --mount=type=secret,id=github_token \ RUN --mount=type=secret,id=github_token \
cd /home/sdp && \ cd /home/sdp && \
docs/platforms/xpu.md
+1 -1
View File
@@ -30,7 +30,7 @@ conda create -n sgl-xpu python=3.12 -y
conda activate sgl-xpu conda activate sgl-xpu
# Set PyTorch XPU as primary pip install channel to avoid installing the larger CUDA-enabled version and prevent potential runtime issues. # Set PyTorch XPU as primary pip install channel to avoid installing the larger CUDA-enabled version and prevent potential runtime issues.
pip3 install torch==2.8.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.4.0 --index-url https://download.pytorch.org/whl/xpu pip3 install torch==2.9.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.5.0 --index-url https://download.pytorch.org/whl/xpu
pip3 install xgrammar --no-deps # xgrammar will introduce CUDA-enabled triton which might conflict with XPU pip3 install xgrammar --no-deps # xgrammar will introduce CUDA-enabled triton which might conflict with XPU
# Clone the SGLang code # Clone the SGLang code
python/pyproject_xpu.toml
+2 -2
View File
@@ -15,8 +15,8 @@ classifiers = [
] ]
dependencies = [ dependencies = [
"torch==2.8.0", "torch==2.9.0",
"torchaudio==2.8.0", "torchaudio==2.9.0",
"torchvision", "torchvision",
"sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git", "sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git",
"IPython", "IPython",