[AMD] Add AMD CI registration (1-gpu unit test) to nightly CI. (#16941)

This commit is contained in:
Michael
2026-01-13 23:41:45 -08:00
committed by GitHub
parent 030496eb06
commit b025cff441
39 changed files with 3701 additions and 2180 deletions
@@ -20,9 +20,10 @@ Tests LRU and FIFO eviction behavior.
import unittest
from sglang.srt.lora.eviction_policy import get_eviction_policy
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=200, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=200, suite="nightly-amd-1-gpu", nightly=True)
class TestLoRAEvictionPolicy(unittest.TestCase):
+3 -2
View File
@@ -9,10 +9,11 @@ import unittest
from unittest.mock import MagicMock
from sglang.srt.entrypoints.openai.serving_base import OpenAIServingBase
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.srt.server_args import ServerArgs
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=30, suite="nightly-1-gpu", nightly=True)
from sglang.srt.server_args import ServerArgs
register_amd_ci(est_time=30, suite="nightly-amd-1-gpu", nightly=True)
class MockTokenizerManager:
@@ -17,9 +17,10 @@ import unittest
import openai
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=150, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=150, suite="nightly-amd-1-gpu", nightly=True)
from sglang.srt.utils import kill_process_tree
from sglang.test.test_utils import (
@@ -17,12 +17,12 @@ import unittest
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.lora_utils import CI_MULTI_LORA_MODELS, run_lora_test_one_by_one
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=200, suite="nightly-1-gpu", nightly=True)
from sglang.test.test_utils import CustomTestCase
register_amd_ci(est_time=200, suite="nightly-amd-1-gpu", nightly=True)
PROMPTS = [
"AI is a field of computer science focused on",