From 62136073f9b5bb26001cb725e955fe9b5e5a2069 Mon Sep 17 00:00:00 2001
From: MingxuZh <109504044+MingxuZh@users.noreply.github.com>
Date: Wed, 29 Apr 2026 19:13:08 -0700
Subject: [PATCH] pin the version of xgrammar to v0.1.32 (#24010)

Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
---
 .github/workflows/pr-test-xpu.yml |  3 ---
 docker/xpu.Dockerfile             | 16 +++++++++-------
 python/pyproject_xpu.toml         |  1 -
 3 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/pr-test-xpu.yml b/.github/workflows/pr-test-xpu.yml
index f820e99ee..aa0554c72 100644
--- a/.github/workflows/pr-test-xpu.yml
+++ b/.github/workflows/pr-test-xpu.yml
@@ -79,9 +79,6 @@ jobs:
           fetch-depth: 0
           ref: ${{ inputs.ref || github.ref }}
 
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-
       - name: Build Docker image
         run: |
           PR_REPO=${{ github.event.pull_request.head.repo.clone_url }}
diff --git a/docker/xpu.Dockerfile b/docker/xpu.Dockerfile
index f69b2cb9e..feec566bb 100644
--- a/docker/xpu.Dockerfile
+++ b/docker/xpu.Dockerfile
@@ -23,12 +23,14 @@ RUN useradd -m -d /home/sdp -s /bin/bash sdp && \
 USER root
 
 # Install the latest UMD driver for SYCL-TLA
-RUN apt-get install -y software-properties-common && \
+RUN apt-get update && apt-get install -y software-properties-common && \
     add-apt-repository -y ppa:kobuk-team/intel-graphics && \
-    apt-get update  && \
-    apt-get install -y libze-intel-gpu1 libze1 intel-metrics-discovery intel-opencl-icd clinfo intel-gsc && \
-    apt-get install -y intel-media-va-driver-non-free libmfx-gen1 libvpl2 libvpl-tools libva-glx2 va-driver-all vainfo && \
-    apt-get install -y libze-dev intel-ocloc
+    apt-get update && \
+    apt-get install -y \
+        libze-intel-gpu1 libze1 intel-metrics-discovery intel-opencl-icd clinfo intel-gsc \
+        intel-media-va-driver-non-free libmfx-gen1 libvpl2 libvpl-tools libva-glx2 va-driver-all vainfo \
+        libze-dev intel-ocloc && \
+    rm -rf /var/lib/apt/lists/*
 
 # Switch to non-root user 'sdp'
 USER sdp
@@ -59,11 +61,11 @@ RUN --mount=type=secret,id=github_token \
     . /home/sdp/miniforge3/bin/activate && \
     conda activate py${PYTHON_VERSION} && \
     echo "Cloning ${SG_LANG_BRANCH} from ${SG_LANG_REPO}" && \
-    git clone --branch ${SG_LANG_BRANCH} --single-branch ${SG_LANG_REPO} && \
+    git clone --branch ${SG_LANG_BRANCH} --single-branch ${SG_LANG_REPO} sglang && \
     cd sglang && cd python && \
     cp pyproject_xpu.toml pyproject.toml && \
     pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \
-    pip install xgrammar --no-deps && \
+    pip install --no-deps xgrammar==0.1.33 && \
     pip install msgspec blake3 py-cpuinfo compressed_tensors gguf partial_json_parser einops tabulate --root-user-action=ignore && \
     conda install libsqlite=3.48.0 -y && \
     # Add environment setup commands to .bashrc again (in case it was overwritten)
diff --git a/python/pyproject_xpu.toml b/python/pyproject_xpu.toml
index 5f2dfbd01..b95017919 100644
--- a/python/pyproject_xpu.toml
+++ b/python/pyproject_xpu.toml
@@ -66,7 +66,6 @@ dependencies = [
   "transformers==5.6.0",
   "uvicorn",
   "uvloop",
-  # "xgrammar==0.1.24", , xgrammar depends on CUDA PyTorch and Triton only
   "smg-grpc-servicer>=0.5.0",
 ]