ci: run Stage A CUDA tests as stage-a-test-small-1-gpu on 5090 (#20988)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user