Skip to content

Fix projection accuracy, deepep support and add scheduler comparison#644

Open
araina-amd wants to merge 8 commits intomainfrom
araina/dev/projection-deepep-fix-and-scheduler-all
Open

Fix projection accuracy, deepep support and add scheduler comparison#644
araina-amd wants to merge 8 commits intomainfrom
araina/dev/projection-deepep-fix-and-scheduler-all

Conversation

@araina-amd
Copy link
Copy Markdown
Contributor

Fix projection accuracy, deepep support and add scheduler comparison

root and others added 6 commits April 1, 2026 09:43
… scheduler comparison

Performance projection fixes:
- Fix double-counting of DeepEP A2A overlap when EP is unchanged
- Correctly reconstruct sequential compute time when EP changes with DeepEP ON
- Fix VPP handling: use interleaved_1f1b when zero-bubble + VPP>1

Scheduler comparison (--pipeline-schedule-algorithm):
- Thread scheduler_algorithm from CLI through projection engine
- Add zbv-formatted and zbv-greedy as CLI choices
- Add _print_scheduler_comparison for multi-scheduler results table
- 'all' mode runs all applicable schedulers + SeaAILab ILP and picks best

CLI fixes:
- Re-add --pipeline-schedule-algorithm argument with full choices
- Rename megatron-ilp to seaailab-ilp
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants