ci: run Stage A CUDA tests as stage-a-test-small-1-gpu on 5090 (#20988)

This commit is contained in:
Lianmin Zheng
2026-03-20 02:55:16 -07:00
committed by GitHub
parent fa89d152c0
commit c1da420799
12 changed files with 78 additions and 100 deletions
+1 -1
View File
@@ -21,7 +21,7 @@ from sglang.srt.utils import get_device
from sglang.test.ci.ci_register import register_amd_ci
from sglang.test.test_utils import CustomTestCase
register_amd_ci(est_time=2, suite="stage-a-test-1-amd")
register_amd_ci(est_time=2, suite="stage-a-test-small-1-gpu-amd")
device = get_device()
@@ -7,7 +7,7 @@ import torch.nn.functional as F
from sglang.test.ci.ci_register import register_amd_ci
from sglang.test.test_utils import CustomTestCase
register_amd_ci(est_time=10, suite="stage-a-test-1-amd")
register_amd_ci(est_time=10, suite="stage-a-test-small-1-gpu-amd")
def _fp8_available() -> bool: