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

Use level-1 (quiet) busy memory check in chunked-prefill and streaming tests (#27303)

Browse Source
This commit is contained in:
Liangsheng Yin
2026-06-04 17:04:27 -04:00
committed by GitHub
parent 687cfe9198
commit cd98d97037
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/test/server_fixtures/streaming_session_fixture.py
+2 -2
View File
@@ -402,7 +402,7 @@ class StreamingSessionServerBase(CustomTestCase):
- `env_overrides`: list of `(env_attr_name, value)` tuples; each is - `env_overrides`: list of `(env_attr_name, value)` tuples; each is
pushed onto the `setUpClass` context stack so the env override is pushed onto the `setUpClass` context stack so the env override is
live during `popen_launch_server` and torn down on live during `popen_launch_server` and torn down on
`tearDownClass`-time. `SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY=2` `tearDownClass`-time. `SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY=1`
is always applied on top of these. is always applied on top of these.
""" """
@@ -417,7 +417,7 @@ class StreamingSessionServerBase(CustomTestCase):
with contextlib.ExitStack() as stack: with contextlib.ExitStack() as stack:
stack.enter_context( stack.enter_context(
envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(2) envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(1)
) )
for name, val in cls.env_overrides: for name, val in cls.env_overrides:
stack.enter_context(getattr(envs, name).override(val)) stack.enter_context(getattr(envs, name).override(val))
test/registered/scheduler/test_mixed_chunked_prefill.py
+1 -1
View File
@@ -29,7 +29,7 @@ class TestMixedChunkedPrefill(GSM8KMixin, CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
with envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(2): with envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(1):
cls.process = popen_launch_server( cls.process = popen_launch_server(
cls.model, cls.model,
cls.base_url, cls.base_url,