[Spec] Redo: split init_backends; account draft weights in --mem-fraction-static (#28855)

Co-authored-by: cctry <cctry@fb.com>
This commit is contained in:
cctry
2026-06-21 20:45:16 -07:00
committed by GitHub
co-authored by cctry
parent fd7874d11b
commit 0c065671c9
18 changed files with 167 additions and 106 deletions
@@ -53,7 +53,7 @@ class TestDeepseekV32(unittest.TestCase):
"--speculative-num-steps=3",
"--speculative-eagle-topk=1",
"--speculative-num-draft-tokens=4",
"--mem-frac=0.7",
"--mem-frac=0.85",
]
variants = [
# Variant: "dp" - Standard TP=8 + DP=8 with dp-attention
@@ -43,7 +43,7 @@ class TestDeepseekV32CPInSeqSplit(CustomTestCase):
"--speculative-num-draft-tokens",
"4",
"--mem-frac",
"0.7",
"0.85",
"--cuda-graph-max-bs",
"32",
"--max-running-requests",
@@ -109,7 +109,7 @@ class TestDeepseekV32CPRoundRobinSplit(CustomTestCase):
"--speculative-num-draft-tokens",
"4",
"--mem-frac",
"0.7",
"0.85",
"--cuda-graph-max-bs",
"32",
"--max-running-requests",
@@ -31,7 +31,7 @@ _OTHER_ARGS = [
'{"enable_multithread_load": true, "num_threads": 64}',
]
if not is_in_amd_ci():
_OTHER_ARGS += ["--mem-frac", "0.7"]
_OTHER_ARGS += ["--mem-frac", "0.85"]
class TestDeepseekV3MTP(GSM8KMixin, DefaultServerBase):
@@ -19,7 +19,7 @@ class TestDeepseekV32DPMTP(
DsaMtpServerBase, DsaMtpEvalConfigDefaults, GSM8KMixin, SpecDecodingMixin
):
model = "deepseek-ai/DeepSeek-V3.2"
mem_fraction_static = 0.7
mem_fraction_static = 0.85
enable_dp_attention = True
bs_1_speed_thres = 90
@@ -19,7 +19,7 @@ class TestDeepseekV32TPMTP(
DsaMtpServerBase, DsaMtpEvalConfigDefaults, GSM8KMixin, SpecDecodingMixin
):
model = "deepseek-ai/DeepSeek-V3.2"
mem_fraction_static = 0.7
mem_fraction_static = 0.85
bs_1_speed_thres = 180
@@ -200,7 +200,7 @@ class TestUnifiedDeepSeekV4FlashEagleHiCacheL3(AccuracyTwoPassMixin, CustomTestC
"--chunked-prefill-size",
"8192",
"--mem-fraction-static",
"0.9",
"0.95",
"--disable-shared-experts-fusion",
"--enable-hierarchical-cache",
"--hicache-ratio",