[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:
co-authored by
yctseng0211
parent
68e8d0f68d
commit
98096b5e02
@@ -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=96, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=300, suite="stage-b-test-small-1-gpu-amd")
|
||||
register_cuda_ci(est_time=96, suite="stage-b-test-small-1-gpu-5090")
|
||||
|
||||
import time
|
||||
|
||||
@@ -1,7 +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=524, suite="stage-b-test-small-1-gpu")
|
||||
|
||||
register_amd_ci(est_time=524, suite="stage-b-test-small-1-gpu-amd")
|
||||
"""
|
||||
Consolidated HiCache variant tests.
|
||||
Tests HiCache with different configurations: standard, MLA, EAGLE, and page size variants.
|
||||
@@ -114,6 +114,7 @@ class TestHiCacheMLA(HiCacheBaseServer, HiCacheEvalMixin, HiCacheMGSMEvalMixin):
|
||||
expected_mmlu_score = 0.5
|
||||
|
||||
|
||||
@unittest.skipIf(is_hip(), "Disabled for AMD-aiter")
|
||||
class TestHiCacheEagle(HiCacheBaseServer, HiCacheEvalMixin):
|
||||
"""HiCache with EAGLE speculative decoding tests"""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user