[ci hot fix] fix global server args init for mamba ut (#16821)

This commit is contained in:
Liangsheng Yin
2026-01-10 02:03:32 +08:00
committed by GitHub
parent bd1afeb568
commit 7066711529
@@ -11,6 +11,7 @@ from sglang.srt.mem_cache.mamba_radix_cache import MambaRadixCache
from sglang.srt.mem_cache.memory_pool import HybridLinearKVPool, HybridReqToTokenPool
from sglang.srt.mem_cache.radix_cache import RadixKey
from sglang.srt.sampling.sampling_params import SamplingParams
from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
@@ -130,6 +131,9 @@ class TestMamba(unittest.TestCase):
assert req_to_token_pool.mamba_pool.available_size() == mamba_cache_size - 1
def test_mamba_radix_cache_1(self):
set_global_server_args_for_scheduler(
ServerArgs(model_path="dummy", page_size=1)
)
# kv cache
size = 128
dtype = torch.bfloat16