From 633929555614953024451927e2a29b78625e6afa Mon Sep 17 00:00:00 2001 From: jiayisunx Date: Fri, 22 May 2026 16:09:08 +0800 Subject: [PATCH] [XPU] add apache-tvm-ffi dependency (#26053) --- docker/xpu.Dockerfile | 1 + docs_new/docs/hardware-platforms/xpu.mdx | 1 + 2 files changed, 2 insertions(+) diff --git a/docker/xpu.Dockerfile b/docker/xpu.Dockerfile index 1a3de356b..832df26f3 100644 --- a/docker/xpu.Dockerfile +++ b/docker/xpu.Dockerfile @@ -68,6 +68,7 @@ RUN --mount=type=secret,id=github_token \ cp pyproject_xpu.toml pyproject.toml && \ pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \ pip install --no-deps xgrammar==0.1.33 && \ + pip install apache-tvm-ffi && \ pip install triton-xpu==3.7.1 --index-url https://download.pytorch.org/whl/xpu --force-reinstall && \ # Add environment setup commands to .bashrc again (in case it was overwritten) echo ". /home/sdp/miniforge3/bin/activate; conda activate py${PYTHON_VERSION}; cd /home/sdp" >> /home/sdp/.bashrc diff --git a/docs_new/docs/hardware-platforms/xpu.mdx b/docs_new/docs/hardware-platforms/xpu.mdx index b5a6c8a2c..4ec4d4253 100644 --- a/docs_new/docs/hardware-platforms/xpu.mdx +++ b/docs_new/docs/hardware-platforms/xpu.mdx @@ -55,6 +55,7 @@ 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. pip3 install torch==2.11.0+xpu torchao torchvision torchaudio==2.11.0+xpu --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 apache-tvm-ffi # xgrammar requires apache-tvm-ffi # Clone the SGLang code git clone https://github.com/sgl-project/sglang.git