minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 17 Packages Projects Releases Wiki Activity

[AMD] Run jit kernel PR test through run_suite.py register mechanism (#24987)

Browse Source
This commit is contained in:
YC Yen-Ching Tseng
2026-05-13 02:57:08 -07:00
committed by GitHub
parent fc20f5b114
commit cf92ccbf18
4 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-amd-rocm720.yml
+1 -1
View File
@@ -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]
.github/workflows/pr-test-amd.yml
+1 -1
View File
@@ -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.
python/sglang/jit_kernel/tests/test_store_cache.py
+2 -1
View File
@@ -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)]
test/run_suite.py
+1
View File
@@ -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",