[AMD] ci: register 5 framework tests to run on AMD CI (#25208)

Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
Michael
2026-05-16 22:56:57 -07:00
committed by GitHub
co-authored by HAI
parent c1d9e37a52
commit 4ef9bad223
5 changed files with 26 additions and 7 deletions
@@ -13,9 +13,9 @@ import unittest
import aiohttp
import requests
from sglang.srt.utils import kill_process_tree
from sglang.srt.utils import is_hip, kill_process_tree
from sglang.srt.utils.hf_transformers_utils import get_tokenizer
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 (
DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -25,6 +25,7 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=87, stage="extra-a", runner_config="1-gpu-large")
register_amd_ci(est_time=87, suite="stage-b-test-1-gpu-large-amd")
def remove_prefix(text: str, prefix: str) -> str:
@@ -550,6 +551,14 @@ class TestSessionControl(CustomTestCase):
outputs_from_session == outputs_normal
), f"outputs_from_session: {outputs_from_session}, outputs_normal: {outputs_normal}"
@unittest.skipIf(
is_hip(),
"Session-branching produces a deterministic 1-token divergence from "
"plain generation on AMD (greedy temperature=0). The other 4 of 5 "
"branch outputs match exactly; passes on CUDA. Suspected ROCm "
"numerical-precision difference in the session KV-cache reuse path. "
"Re-enable when that divergence is fixed.",
)
def test_session_control_with_branching(self):
root_prompt = "First, let me explain in one sentence about AI"
chunks_per_step = [