[AMD CI] migrate and re-enable CI tests to new CI registry (#16949)

Co-authored-by: yctseng0211 <yctseng@amd.com>
This commit is contained in:
Bingxu Chen
2026-01-14 21:25:25 -08:00
committed by GitHub
co-authored by yctseng0211
parent 68e8d0f68d
commit 98096b5e02
26 changed files with 86 additions and 40 deletions
+6 -2
View File
@@ -1,6 +1,7 @@
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=181, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=330, suite="stage-b-test-small-1-gpu-amd")
import unittest
from types import SimpleNamespace
@@ -62,7 +63,10 @@ class TestLLaDA2Mini(CustomTestCase):
print(f"{metrics=}")
self.assertGreater(metrics["accuracy"], 0.88)
self.assertGreater(metrics["output_throughput"], 150)
if is_in_amd_ci():
self.assertGreater(metrics["output_throughput"], 80)
else:
self.assertGreater(metrics["output_throughput"], 150)
def test_bs_1_speed(self):
args = BenchArgs(port=int(self.base_url.split(":")[-1]), max_new_tokens=2048)