[AMD] CI - add partitions for stage-b-test-small-1-gpu-amd (#17345)

This commit is contained in:
YC Tseng
2026-01-19 08:16:07 -08:00
committed by GitHub
parent 2ea02f0642
commit 1a053a810c
5 changed files with 8 additions and 8 deletions
@@ -16,7 +16,7 @@ from sglang.test.test_utils import (
# Sliding window attention with Triton backend (Gemma-3 model)
register_cuda_ci(est_time=100, suite="stage-b-test-large-1-gpu")
register_amd_ci(est_time=100, suite="stage-b-test-small-1-gpu-amd")
register_amd_ci(est_time=200, suite="stage-b-test-small-1-gpu-amd")
class TestSlidingWindowAttentionTriton(CustomTestCase):