From 64812ebe10f29b7a76defe457498f9f00408a69c Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Mon, 27 Apr 2026 17:23:54 +0800 Subject: [PATCH 1/2] Update test_gpt_oss_1gpu.py --- test/registered/core/test_gpt_oss_1gpu.py | 31 +++++++++++++++-------- 1 file changed, 21 insertions(+), 10 deletions(-) diff --git a/test/registered/core/test_gpt_oss_1gpu.py b/test/registered/core/test_gpt_oss_1gpu.py index c6b6e5162b12..96f5a30a3cda 100644 --- a/test/registered/core/test_gpt_oss_1gpu.py +++ b/test/registered/core/test_gpt_oss_1gpu.py @@ -1,23 +1,34 @@ +import os import unittest +from sglang.srt.utils import is_hip from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci from sglang.test.gpt_oss_common import BaseTestGptOss -register_cuda_ci(est_time=408, suite="stage-b-test-1-gpu-large") +register_cuda_ci(est_time=480, suite="stage-b-test-1-gpu-large") register_amd_ci(est_time=750, suite="stage-b-test-1-gpu-small-amd-mi35x") class TestGptOss1Gpu(BaseTestGptOss): def test_mxfp4_20b(self): - self.run_test( - model_variant="20b", - quantization="mxfp4", - expected_score_of_reasoning_effort={ - "low": 0.34, - "medium": 0.34, - "high": 0.27, # TODO investigate - }, - ) + prev = os.environ.get("SGLANG_USE_AITER") + if is_hip(): + os.environ["SGLANG_USE_AITER"] = "1" + try: + self.run_test( + model_variant="20b", + quantization="mxfp4", + expected_score_of_reasoning_effort={ + "low": 0.34, + "medium": 0.34, + "high": 0.27, # TODO investigate + }, + ) + finally: + if prev is None: + os.environ.pop("SGLANG_USE_AITER", None) + else: + os.environ["SGLANG_USE_AITER"] = prev def test_bf16_20b(self): self.run_test( From ffa5c7b07d23b75cbee0885f5f2d4e6777a5e9f8 Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Mon, 27 Apr 2026 17:24:56 +0800 Subject: [PATCH 2/2] Update test_gpt_oss_1gpu.py --- test/registered/core/test_gpt_oss_1gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/registered/core/test_gpt_oss_1gpu.py b/test/registered/core/test_gpt_oss_1gpu.py index 96f5a30a3cda..11f3d186da76 100644 --- a/test/registered/core/test_gpt_oss_1gpu.py +++ b/test/registered/core/test_gpt_oss_1gpu.py @@ -5,7 +5,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci from sglang.test.gpt_oss_common import BaseTestGptOss -register_cuda_ci(est_time=480, suite="stage-b-test-1-gpu-large") +register_cuda_ci(est_time=408, suite="stage-b-test-1-gpu-large") register_amd_ci(est_time=750, suite="stage-b-test-1-gpu-small-amd-mi35x")