From 0a304870e885e723fa5d56bfc6cc713df4b33b56 Mon Sep 17 00:00:00 2001 From: Zaili Wang <109502517+ZailiWang@users.noreply.github.com> Date: Sun, 12 Oct 2025 05:08:28 +0800 Subject: [PATCH] fix Xeon CI (#11454) --- .github/workflows/pr-test-xeon.yml | 16 ++++------------ docker/Dockerfile.xeon | 2 +- 2 files changed, 5 insertions(+), 13 deletions(-) diff --git a/.github/workflows/pr-test-xeon.yml b/.github/workflows/pr-test-xeon.yml index 358dc3192..75a955a08 100644 --- a/.github/workflows/pr-test-xeon.yml +++ b/.github/workflows/pr-test-xeon.yml @@ -41,10 +41,12 @@ jobs: run: | version=$(cat python/sglang/version.py | cut -d'"' -f2) tag=v${version}-xeon + PR_REPO=${{ github.event.pull_request.head.repo.clone_url }} + PR_HEAD_REF=${{ github.head_ref }} docker build \ - --build-arg SGLANG_REPO=${{ github.event.pull_request.head.repo.clone_url }} \ - --build-arg VER_SGLANG=${{ github.head_ref }} \ + ${PR_REPO:+--build-arg SGLANG_REPO=$PR_REPO} \ + ${PR_HEAD_REF:+--build-arg VER_SGLANG=$PR_HEAD_REF} \ . -f docker/Dockerfile.xeon -t sglang_xeon --no-cache - name: Run container @@ -55,15 +57,6 @@ jobs: --name ci_sglang_xeon \ sglang_xeon - - name: Install dependencies - timeout-minutes: 20 - run: | - docker exec ci_sglang_xeon bash -c "python3 -m pip install --upgrade pip" - docker exec ci_sglang_xeon pip uninstall sgl-kernel -y || true - docker exec -w /sglang-checkout/sgl-kernel ci_sglang_xeon bash -c "cp pyproject_cpu.toml pyproject.toml && pip install -v ." - docker exec -w /sglang-checkout/ ci_sglang_xeon bash -c "rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml" - docker exec -w /sglang-checkout/ ci_sglang_xeon bash -c "pip install -e "python[dev_cpu]"" - - name: Check AMX support id: check_amx timeout-minutes: 5 @@ -72,7 +65,6 @@ jobs: bash -c "python3 -c 'import torch; import sgl_kernel; assert torch._C._cpu._is_amx_tile_supported(); assert hasattr(torch.ops.sgl_kernel, \"convert_weight_packed\"); '" - name: Run unit tests - if: steps.check_amx.outcome == 'success' timeout-minutes: 36 run: | docker exec -w /sglang-checkout/ ci_sglang_xeon \ diff --git a/docker/Dockerfile.xeon b/docker/Dockerfile.xeon index 9f842d515..17572397d 100644 --- a/docker/Dockerfile.xeon +++ b/docker/Dockerfile.xeon @@ -35,7 +35,7 @@ ENV CONDA_PREFIX=/sgl-workspace/miniforge3 RUN pip config set global.index-url https://download.pytorch.org/whl/cpu && \ pip config set global.extra-index-url https://pypi.org/simple -RUN git clone ${SGLANG_REPO} && \ +RUN git clone ${SGLANG_REPO} sglang && \ cd sglang && \ git checkout ${VER_SGLANG} && \ cd python && \