Skip to content

Reland spec v2 tree drafting (eagle topk>1) with page_size==1 (#26866)#26997

Merged
hnyls2002 merged 16 commits into
mainfrom
lsyin/reland-26866
Jun 3, 2026
Merged

Reland spec v2 tree drafting (eagle topk>1) with page_size==1 (#26866)#26997
hnyls2002 merged 16 commits into
mainfrom
lsyin/reland-26866

Conversation

@hnyls2002

@hnyls2002 hnyls2002 commented Jun 2, 2026

Copy link
Copy Markdown
Collaborator

Relands #26866 (reverted in #26981).


CI States

Latest PR Test (Base): 🚫 Run #26907731073
Latest PR Test (Extra): ✅ Run #26907730270

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@hnyls2002

Copy link
Copy Markdown
Collaborator Author

/tag-and-rerun-ci extra

@hnyls2002

Copy link
Copy Markdown
Collaborator Author

/rerun-test test_spec_eagle_page.py test_eagle_dp_attention.py test_eagle_infer_beta_dp_attention_large.py

@hnyls2002

Copy link
Copy Markdown
Collaborator Author

/rerun-test test_constrained_decoding_spec_reasoning.py

@github-actions

github-actions Bot commented Jun 2, 2026

Copy link
Copy Markdown
Contributor

Results for /rerun-test test_constrained_decoding_spec_reasoning.py:

🚀 2-gpu-h100 (1 test): ✅ View workflow run

cd test/ && python3 registered/spec/test_constrained_decoding_spec_reasoning.py

@hnyls2002

Copy link
Copy Markdown
Collaborator Author

/rerun-test test_qwen3_next_models_mtp.py

@github-actions

github-actions Bot commented Jun 2, 2026

Copy link
Copy Markdown
Contributor

Results for /rerun-test test_qwen3_next_models_mtp.py:

🚀 4-gpu-h100 (1 test): ✅ View workflow run

cd test/ && python3 registered/models_e2e/test_qwen3_next_models_mtp.py

@sgl-project sgl-project deleted a comment from github-actions Bot Jun 3, 2026
@hnyls2002

Copy link
Copy Markdown
Collaborator Author

Latest PR Test (Base): ✅ Run #26898846215
Latest PR Test (Extra): ❌ Run #26898843678

@hnyls2002
hnyls2002 merged commit ac99794 into main Jun 3, 2026
70 of 146 checks passed
@hnyls2002
hnyls2002 deleted the lsyin/reland-26866 branch June 3, 2026 19:40
Oasis-Git added a commit to Oasis-Git/sglang that referenced this pull request Jun 3, 2026
…acy writes

Conflicts:

- python/sglang/srt/speculative/eagle_draft_extend_cuda_graph_runner.py:
  PR sgl-project#26997 (Reland spec v2 tree drafting for eagle topk>1 with
  page_size==1) resizes the cuda-graph buffer width from
  speculative_num_steps + 1 to speculative_num_draft_tokens. Take
  upstream — this is a correctness fix for topk > 1 draft-extend
  buffer overflow.

Silent legacy writes flagged by post-merge sweep (NOT introduced by
this merge — pre-existing breakage in cg-refactor's earlier disable_
cuda_graph migration; surfaced now while we have the audit running):

- python/sglang/compile_deep_gemm.py refine_server_args() and
  python/sglang/srt/arg_groups/pd_disaggregation_hook.py both wrote
  server_args.disable_cuda_graph = True after ServerArgs.__post_init__
  had already consumed that field. The post_init read-once-then-mirror
  pattern means those late writes were silent no-ops. Switch both to
  set cuda_graph_config[Phase.DECODE].backend / [Phase.PREFILL].backend
  = Backend.DISABLED directly — same semantic, propagates correctly.

Validation: attention unittests 181 passed / 0 failures / 548 subtests.
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.

2 participants