minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD CI] migrate and re-enable CI tests to new CI registry (#16949)

Browse Source 
Co-authored-by: yctseng0211 <yctseng@amd.com>
This commit is contained in:
Bingxu Chen
2026-01-14 21:25:25 -08:00
committed by GitHub
co-authored by yctseng0211
parent 68e8d0f68d
commit 98096b5e02
26 changed files with 86 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/core/test_hidden_states.py
+9 -1
View File
@@ -4,12 +4,20 @@ import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
import sglang as sgl
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.srt.utils import is_hip
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import DEFAULT_SMALL_MODEL_NAME_FOR_TEST, CustomTestCase
register_cuda_ci(est_time=55, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=55, suite="stage-b-test-small-1-gpu-amd")
register_cuda_ci(est_time=31, suite="stage-b-test-small-1-gpu-5090")
_is_hip = is_hip()
if _is_hip:
import os
os.environ["SGLANG_USE_AITER"] = "0"
class TestHiddenState(CustomTestCase):
def test_return_hidden_states(self):