[AMD] Add AMD CI registration (1-gpu unit test) to nightly CI. (#16941)
This commit is contained in:
@@ -9,10 +9,11 @@ import torch
|
||||
import torch.distributed as dist
|
||||
import torch.multiprocessing as mp
|
||||
|
||||
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.test_utils import CustomTestCase
|
||||
|
||||
register_cuda_ci(est_time=30, suite="nightly-2-gpu", nightly=True)
|
||||
register_amd_ci(est_time=60, suite="nightly-amd", nightly=True)
|
||||
|
||||
|
||||
class TestDumperPureFunctions(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user