From cf92ccbf1871235cdf586f627b11caec565e1ca7 Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Wed, 13 May 2026 17:57:08 +0800 Subject: [PATCH] [AMD] Run jit kernel PR test through run_suite.py register mechanism (#24987) --- .github/workflows/pr-test-amd-rocm720.yml | 2 +- .github/workflows/pr-test-amd.yml | 2 +- python/sglang/jit_kernel/tests/test_store_cache.py | 3 ++- test/run_suite.py | 1 + 4 files changed, 5 insertions(+), 3 deletions(-) diff --git a/.github/workflows/pr-test-amd-rocm720.yml b/.github/workflows/pr-test-amd-rocm720.yml index 990b7dccd..2163e33dc 100644 --- a/.github/workflows/pr-test-amd-rocm720.yml +++ b/.github/workflows/pr-test-amd-rocm720.yml @@ -327,7 +327,7 @@ jobs: - name: Run JIT kernel unit tests timeout-minutes: 10 run: | - bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout" python3 -m pytest -q python/sglang/jit_kernel/tests/test_store_cache.py + bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite jit-kernel-unit-test-amd ${{ inputs.continue_on_error && '--continue-on-error' || '' }} stage-b-test-1-gpu-small-amd-rocm720: needs: [check-changes] diff --git a/.github/workflows/pr-test-amd.yml b/.github/workflows/pr-test-amd.yml index acb9281e3..007ff8b22 100644 --- a/.github/workflows/pr-test-amd.yml +++ b/.github/workflows/pr-test-amd.yml @@ -355,7 +355,7 @@ jobs: - name: Run JIT kernel unit tests timeout-minutes: 10 run: | - bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout" python3 -m pytest -q python/sglang/jit_kernel/tests/test_store_cache.py + bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite jit-kernel-unit-test-amd ${{ needs.check-changes.outputs.continue_on_error == 'true' && '--continue-on-error' || '' }} # =============================================== Wait Jobs for Sequential PR Execution ==================================================== # These jobs poll GitHub API to wait for previous stages to complete. diff --git a/python/sglang/jit_kernel/tests/test_store_cache.py b/python/sglang/jit_kernel/tests/test_store_cache.py index 278781ca2..a7b9e5be1 100644 --- a/python/sglang/jit_kernel/tests/test_store_cache.py +++ b/python/sglang/jit_kernel/tests/test_store_cache.py @@ -6,10 +6,11 @@ import torch from sglang.jit_kernel.kvcache import can_use_store_cache, store_cache from sglang.jit_kernel.utils import get_ci_test_range -from sglang.test.ci.ci_register import register_cuda_ci +from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci register_cuda_ci(est_time=28, suite="stage-b-kernel-unit-1-gpu-large") register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True) +register_amd_ci(est_time=55, suite="jit-kernel-unit-test-amd") BS_LIST = [2**n for n in range(0, 15)] BS_LIST += [x + 1 + i for i, x in enumerate(BS_LIST)] diff --git a/test/run_suite.py b/test/run_suite.py index c09b1fae8..597fe3c9b 100644 --- a/test/run_suite.py +++ b/test/run_suite.py @@ -32,6 +32,7 @@ PER_COMMIT_SUITES = { "stage-b-test-large-8-gpu-35x-disaggregation-amd", "stage-b-test-1-gpu-large-amd", "stage-b-test-2-gpu-large-amd", + "jit-kernel-unit-test-amd", "stage-c-test-4-gpu-amd", "stage-c-test-large-8-gpu-amd", "stage-c-test-large-8-gpu-amd-mi35x",