diff --git a/.github/workflows/pr-test-xeon.yml b/.github/workflows/pr-test-xeon.yml index 109c0d4b06ef..59c5767fd03d 100644 --- a/.github/workflows/pr-test-xeon.yml +++ b/.github/workflows/pr-test-xeon.yml @@ -111,11 +111,17 @@ jobs: docker exec -w /sglang-checkout/ ci_sglang_xeon \ bash -c "source /opt/.venv/bin/activate && 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 + #- name: Run custom unit tests + # timeout-minutes: 36 + # run: | + # docker exec -w /sglang-checkout/ ci_sglang_xeon \ + # bash -c "source /opt/.venv/bin/activate && cd ./test/srt && python3 run_suite.py --suite per-commit-cpu --timeout-per-file 1500" + + - name: Run normal unit tests timeout-minutes: 36 run: | docker exec -w /sglang-checkout/ ci_sglang_xeon \ - bash -c "source /opt/.venv/bin/activate && cd ./test/srt && python3 run_suite.py --suite per-commit-cpu --timeout-per-file 1500" + bash -c "source /opt/.venv/bin/activate && cd ./test && python3 run_suite.py --hw cpu --suite stage-b-test-cpu" - name: Change permission timeout-minutes: 2 diff --git a/test/registered/openai_server/basic/test_protocol.py b/test/registered/openai_server/basic/test_protocol.py index b707f35bff76..33341276f370 100644 --- a/test/registered/openai_server/basic/test_protocol.py +++ b/test/registered/openai_server/basic/test_protocol.py @@ -28,10 +28,11 @@ ModelList, UsageInfo, ) -from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci +from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci, register_cpu_ci register_cuda_ci(est_time=2, suite="stage-b-test-1-gpu-small") register_amd_ci(est_time=10, suite="stage-b-test-1-gpu-small-amd") +register_cpu_ci(est_time=10, suite="stage-b-test-cpu") class TestModelCard(unittest.TestCase): diff --git a/test/run_suite.py b/test/run_suite.py index 7fe58ad9167a..bd7e261e5b88 100644 --- a/test/run_suite.py +++ b/test/run_suite.py @@ -23,7 +23,10 @@ # Per-commit test suites (run on every PR) PER_COMMIT_SUITES = { - HWBackend.CPU: ["stage-a-test-cpu"], + HWBackend.CPU: [ + "stage-a-test-cpu", + "stage-b-test-cpu" + ], HWBackend.AMD: [ "stage-a-test-1-gpu-small-amd", "stage-b-test-1-gpu-small-amd",