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

sm120 32GB mem-tier: raise decode cuda-graph max_bs 24->48 + chunked_prefill 2k->4k (#37898)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-09-04 18:47:26 +08:00
committed by GitHub
parent 12735c2d76
commit f3b2725609
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/arg_groups/memory_hook.py
+7 -4
View File
@@ -77,18 +77,21 @@ def handle_gpu_memory_settings(server_args: Any, gpu_mem):
decode_cuda_graph_config.max_bs = 8 decode_cuda_graph_config.max_bs = 8
elif gpu_mem < 35 * 1024: elif gpu_mem < 35 * 1024:
# A10, 4090, 5090 # A10, 4090, 5090
# (chunked_prefill_size 2k, max_bs 24 if tp < 4 else 80) # (chunked_prefill_size 4k, max_bs 48 if tp < 4 else 160)
# 32GB Blackwell (RTX 5090) can hold decode cuda graphs well past
# bs=24; the previous cap forced eager decode at bs>=32 and
# collapsed high-concurrency throughput vs vLLM.
if cfg.chunked_prefill_size is None: if cfg.chunked_prefill_size is None:
declare_resolution( declare_resolution(
server_args, server_args,
"_handle_gpu_memory_settings", "_handle_gpu_memory_settings",
chunked_prefill_size=2048, chunked_prefill_size=4096,
) )
if decode_cuda_graph_config.max_bs is None: if decode_cuda_graph_config.max_bs is None:
if cfg.tp_size < 4: if cfg.tp_size < 4:
decode_cuda_graph_config.max_bs = 24 decode_cuda_graph_config.max_bs = 48
else: else:
decode_cuda_graph_config.max_bs = 80 decode_cuda_graph_config.max_bs = 160
elif gpu_mem < 60 * 1024: elif gpu_mem < 60 * 1024:
# A100 (40GB), L40, # A100 (40GB), L40,
# (chunked_prefill_size 4k, max_bs 32 if tp < 4 else 160) # (chunked_prefill_size 4k, max_bs 32 if tp < 4 else 160)