Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Threshold-Based Exclusive Batching for LLM Inference

About

Mixed batching (MB)--interleaving prefill and decode in a single batch--has become the standard scheduling strategy for large language model (LLM) inference due to its efficiency in maximizing compute and memory utilization. However, through controlled experiments, we find that prefill-decode interference inflates MB's per-step marginal cost above that of pure decode. On the high-bandwidth H200 (4.8 TB/s), this occurs only when decode tokens exceed 80% of the batch; however, on the bandwidth-constrained RTX PRO 6000 (1.792 TB/s), this threshold plummets to just 20%. Consequently, the optimal choice between MB and exclusive batching (EB) fundamentally depends on GPU memory bandwidth, model size, and workload composition. We derive a closed-form condition for this EB-MB performance crossover, along with asymptotically optimal phase-switching thresholds and memory-safe batch sizing for EB. Optimized EB achieves up to 41.9% higher throughput on bandwidth-constrained GPUs, while MB retains its advantage on high-bandwidth hardware with larger models. Our hybrid scheduler EB+ applies this condition online to dynamically switch between EB and MB without manual intervention. Under non-stationary traffic with distribution or concurrency shifts, EB+ attains the highest or near-highest throughput in every setting, outperforming MB by up to 36.4%.

Weifang Zhang, Yuzhou Nie, Bowen Pang, Guangrui Ma, Shining Wu• 2026

Related benchmarks

TaskDatasetResultRank
LLM InferenceQwen3-8B 2k prompts Decode-heavy workload
Throughput (tok/s)4.68e+4
30
LLM InferenceQwen3-8B 2k prompts Balanced workload
Throughput (tok/s)6.10e+4
28
LLM InferenceQwen3-8B 2k prompts Prefill-heavy workload
Throughput (tok/s)1.05e+5
26
LLM InferenceQwen3-8B workload (c=512) on RTX PRO 6000
SLO Attainment (%)80.3
18
LLM InferenceQwen3-8B synthetic workload (mu_L=512, mu_O=256)
Throughput (tok/s)4.80e+4
16
LLM InferenceWildChat
RPS (Requests/s)74.05
11
LLM InferenceShareGPT
Throughput (RPS)58.33
6
LLM InferenceLongBench
Throughput (RPS)96.93
6
LLM InferenceNuminaMath
Throughput (RPS)12.8
6
Showing 9 of 9 rows

Other info

Follow for update