Migrate backends tests to CI registration system (#16468)
This commit is contained in:
@@ -5,6 +5,7 @@ from types import SimpleNamespace
|
||||
import requests
|
||||
|
||||
from sglang.srt.utils import kill_process_tree
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
from sglang.test.run_eval import run_eval
|
||||
from sglang.test.test_utils import (
|
||||
DEFAULT_MODEL_NAME_FOR_TEST,
|
||||
@@ -15,6 +16,9 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=190, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=169, suite="stage-b-test-small-1-gpu")
|
||||
|
||||
|
||||
class TestTorchCompile(CustomTestCase):
|
||||
@classmethod
|
||||
@@ -32,7 +32,6 @@ suites = {
|
||||
TestFile("test_profile_merger.py", 8),
|
||||
TestFile("test_profile_merger_http_api.py", 9),
|
||||
TestFile("test_swa_unittest.py", 8),
|
||||
TestFile("test_torch_compile.py", 190),
|
||||
TestFile("test_utils_update_weights.py", 29),
|
||||
TestFile("test_video_utils.py", 5),
|
||||
TestFile("test_modelopt_export.py", 9),
|
||||
@@ -142,7 +141,6 @@ suite_amd = {
|
||||
TestFile("test_srt_engine.py", 261),
|
||||
TestFile("test_start_profile.py", 60),
|
||||
TestFile("test_swa_unittest.py", 10),
|
||||
TestFile("test_torch_compile.py", 169),
|
||||
# TestFile("test_torch_compile_moe.py", 210), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/13107
|
||||
TestFile("test_type_based_dispatcher.py", 10),
|
||||
TestFile("test_video_utils.py", 8),
|
||||
|
||||
Reference in New Issue
Block a user