[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
@@ -34,13 +34,17 @@ from typing import Any, Dict, List, Optional, Tuple
|
||||
import numpy as np
|
||||
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 HFRunner, SRTRunner
|
||||
|
||||
register_cuda_ci(
|
||||
est_time=150,
|
||||
suite="stage-b-test-small-1-gpu",
|
||||
)
|
||||
register_amd_ci(
|
||||
est_time=250,
|
||||
suite="stage-b-test-small-1-gpu-amd",
|
||||
)
|
||||
register_cuda_ci(
|
||||
est_time=76,
|
||||
suite="stage-b-test-small-1-gpu-5090",
|
||||
|
||||
Reference in New Issue
Block a user