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

[AMD] Restore AMD CI registrations dropped by #37436 (#38581)

Browse Source
This commit is contained in:
Michael
2026-09-09 23:59:10 -07:00
committed by GitHub
parent f3152439d0
commit 026e61bd94
5 changed files with 26 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/debug_utils/test_tensor_dump_forward_hook.py
+4 -1
View File
@@ -14,11 +14,14 @@ from sglang.srt.layers.linear import LinearBase
from sglang.srt.models.qwen2 import Qwen2MLP
from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler
from sglang.srt.utils import add_prefix, get_device
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.layer_ut_utils import init_single_process_dist
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small")
# Backend-specific: the hook resolves get_device() and the matching distributed
# backend, so only an AMD run exercises the HIP device and dump path.
register_amd_ci(est_time=15, suite="stage-b-test-1-gpu-small-amd")
TEST_HIDDEN_SIZE = 32