[AMD CI] Migrate and Add More Testcases (#17116)

Co-authored-by: yctseng0211 <yctseng@amd.com>
This commit is contained in:
Bingxu Chen
2026-01-19 08:07:39 -08:00
committed by GitHub
co-authored by yctseng0211
parent 20b0523eca
commit 2ea02f0642
19 changed files with 311 additions and 67 deletions
@@ -24,7 +24,7 @@ from sglang.srt.layers.attention.wave_ops.prefill_attention import (
from sglang.test.ci.ci_register import register_amd_ci
# Wave attention kernel unit tests (AMD only - requires wave_lang)
register_amd_ci(est_time=60, suite="stage-a-test-1")
register_amd_ci(est_time=60, suite="stage-a-test-1-amd")
class TestWaveAttention(unittest.TestCase):