[Test] Enable mamba lazy extra buffer alloc edge case kl test in CI (#31991)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user