[Test] Enable mamba lazy extra buffer alloc edge case kl test in CI (#31991)

This commit is contained in:
Hanming Lu
2026-07-21 20:45:43 -07:00
committed by GitHub
parent 246b3c3eaf
commit 297f9336c8
@@ -7,7 +7,7 @@ from sglang.test.kits.kl_divergence_kit import KLDivergenceMixin
from sglang.test.kits.prefix_cache_branching_kit import PrefixCacheBranchingMixin
from sglang.test.server_fixtures.default_fixture import DefaultServerBase
register_cuda_ci(est_time=250, stage="base-c", runner_config="4-gpu-h100")
register_cuda_ci(est_time=500, stage="base-c", runner_config="4-gpu-h100")
QWEN3_NEXT_MODEL = "Qwen/Qwen3-Next-80B-A3B-Instruct"
@@ -54,10 +54,7 @@ class TestQwen3NextLazyExtraBufferLargePage(
other_args = _make_args(page_size=2, track_interval=2)
@unittest.skip("Manual-only: forces all lazy preallocs to fail")
class TestQwen3NextLazyExtraBufferAllocFail(
GSM8KMixin, KLDivergenceMixin, PrefixCacheBranchingMixin, DefaultServerBase
):
class TestQwen3NextLazyExtraBufferAllocFail(KLDivergenceMixin, DefaultServerBase):
model = QWEN3_NEXT_MODEL
cache_chunk_size = 64
gsm8k_accuracy_thres = 0.93
@@ -77,9 +74,8 @@ class TestQwen3NextLazyExtraBufferAllocFail(
os.environ.pop("SGLANG_TEST_SKIP_CACHE_HIT_ASSERT", None)
@unittest.skip("Manual-only: forces all lazy preallocs to fail")
class TestQwen3NextLazyExtraBufferLargePageAllocFail(
GSM8KMixin, KLDivergenceMixin, PrefixCacheBranchingMixin, DefaultServerBase
KLDivergenceMixin, DefaultServerBase
):
model = QWEN3_NEXT_MODEL
cache_chunk_size = 64