Skip to content

[Performance] qwen3-next improve causal conv1d in prefill phase#10595

Merged
zhyncs merged 4 commits intosgl-project:mainfrom
liz-badada:prefill_causal_conv1d
Sep 18, 2025
Merged

[Performance] qwen3-next improve causal conv1d in prefill phase#10595
zhyncs merged 4 commits intosgl-project:mainfrom
liz-badada:prefill_causal_conv1d

Conversation

@liz-badada
Copy link
Copy Markdown
Collaborator

@liz-badada liz-badada commented Sep 18, 2025

Motivation

If not target_verify, call causal_conv1d_fwd in sgl_kernel for prefill to eliminate the bubbles; If target_verify, then still call triton causal_conv1d_update for spec decoding.

Modifications

Accuracy Tests

Benchmarking and Profiling (TP=4)

prefill before
image

prefill after
image

Batch Size Input Len Output Len Latency (s) TTFT (s) Remark
512 4096 100 45.75 39.14 before
512 4096 100 38.97 32.28 after
32 32684 100 22.29 21.09 before
32 32684 100 19.15 17.95 after

Checklist

@liz-badada liz-badada marked this pull request as ready for review September 18, 2025 06:53
@liz-badada liz-badada changed the title improve causal conv1d in prefill phase [Performance] improve causal conv1d in prefill phase Sep 18, 2025
@liz-badada liz-badada changed the title [Performance] improve causal conv1d in prefill phase [Performance] qwen3-next improve causal conv1d in prefill phase Sep 18, 2025
@zhyncs zhyncs merged commit fc80966 into sgl-project:main Sep 18, 2025
74 of 77 checks passed
chenxu140 added a commit to ping1jing2/sglang that referenced this pull request Sep 20, 2025
* origin/qwen3: (30 commits)
  chore: bump sgl-kernel 0.3.11 (sgl-project#10630)
  feat: add fused moe config for Qwen3-Next-80B-A3B-Instruct on B200 (sgl-project#10631)
  model support: Sarashina2VisionForCausalLM (sgl-project#10632)
  [Performance] Qwen3-Next: speed up update_mamba_state_after_mtp_verify by 10x; e2e up to 3.54% faster (sgl-project#10586)
  [Performance] Qwen3-Next: replace arange to cached query_start_loc_li… (sgl-project#10553)
  [Feature] Speculative decoding support lookahead (sgl-project#9873)
  refactor: use registry for _get_attention_backend_from_str (sgl-project#10629)
  [router] refactor worker to builder pattern 1/n (sgl-project#10628)
  Garbage collector regression in the online server (sgl-project#10621)
  feat: Add FlexAttention Backend for Efficient Sparse Attention (sgl-project#9947)
  Fix bias handling in TritonMoeQuantInfo within quantization/mxfp4.py (sgl-project#10579)
  [Performance] qwen3-next improve causal conv1d in prefill phase (sgl-project#10595)
  Fix sgl_kernel import failure on devices other than CUDA (sgl-project#10610)
  support qwen3-next-fp8 deepep (sgl-project#10622)
  update deepep version for qwen3-next deepep moe (sgl-project#10624)
  Feat/add heartbeat mechanism for nixl conn (sgl-project#10222)
  [RL] Add destroy process group api (sgl-project#9979)
  fix deepep assert when PD disaggregation == null (sgl-project#8274)
  Scale kkt after reduction (sgl-project#10604)
  [improvement] add average input/output token length for hicache benchmark stats output (sgl-project#10525)
  ...
lifuhuang pushed a commit that referenced this pull request Sep 20, 2025
HanHan009527 pushed a commit to HanHan009527/sglang that referenced this pull request Oct 9, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants