[AMD CI] re-enable testcases missed when migrating ci test files (#16535)

Co-authored-by: michael-amd <michael.zhang@amd.com>
Co-authored-by: yctseng0211 <yctseng@amd.com>
This commit is contained in:
Bingxu Chen
2026-01-07 22:43:48 -08:00
committed by GitHub
co-authored by michael-amd yctseng0211
parent 48b8dcd42e
commit f9c0426692
19 changed files with 47 additions and 24 deletions
+2 -1
View File
@@ -16,7 +16,7 @@ import multiprocessing as mp
import os
import unittest
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.lora_utils import (
ALL_OTHER_MULTI_LORA_MODELS,
CI_MULTI_LORA_MODELS,
@@ -25,6 +25,7 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=82, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=82, suite="stage-b-test-small-1-gpu")
class TestLoRA(CustomTestCase):
+6 -1
View File
@@ -17,7 +17,7 @@ import os
import unittest
from typing import List
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.lora_utils import (
ALL_OTHER_LORA_MODELS,
BACKENDS,
@@ -30,6 +30,11 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=200, suite="stage-b-test-small-1-gpu")
register_amd_ci(
est_time=200,
suite="stage-b-test-small-1-gpu",
disabled="see https://github.com/sgl-project/sglang/issues/13107",
)
class TestLoRABackend(CustomTestCase):
+2 -1
View File
@@ -19,11 +19,12 @@ from typing import Dict, List, Tuple
import torch
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.runners import SRTRunner
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=224, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=224, suite="stage-b-test-small-1-gpu")
PROMPTS = [
"AI is a field of computer science focused on",
+6 -1
View File
@@ -17,7 +17,7 @@ import os
import unittest
from typing import List
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.lora_utils import (
ALL_OTHER_LORA_MODELS,
CI_LORA_MODELS,
@@ -29,6 +29,11 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=116, suite="stage-b-test-large-2-gpu")
register_amd_ci(
est_time=116,
suite="stage-b-test-large-2-gpu-amd",
disabled="see https://github.com/sgl-project/sglang/issues/13107",
)
class TestLoRATP(CustomTestCase):
@@ -16,7 +16,7 @@ import multiprocessing as mp
import os
import unittest
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.lora_utils import (
ALL_OTHER_MULTI_LORA_MODELS,
CI_MULTI_LORA_MODELS,
@@ -25,6 +25,7 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=60, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=60, suite="stage-b-test-small-1-gpu")
# All prompts are used at once in a batch.
PROMPTS = [