Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
araina-amd wants to merge 8 commits into
base: main
Choose a base branch
from
Open

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
2443313
feat(primus-pipeline): fix te gemm patch for zerobubble
Mar 20, 2026
910bcde
feat(primus-pipeline): add lagacy grouped gemm support
ChengYao-amd Mar 26, 2026
187ad1d
feat(primus-pipeline): add turbo fp8 support
ChengYao-amd Mar 30, 2026
64024ca
feat(primus-pipeline): fix newest megatron issue
ChengYao-amd Mar 30, 2026
a7197eb
fix(pp): fix original PP dump data logic
ChengYao-amd Mar 31, 2026
28190e5
Fix projection accuracy, rename Megatron ILP to SeaAILab ILP, and add…
araina-amd Apr 4, 2026
8aaa46e
fix: replace empty except clause with log_rank_0 warning in dump_pp_d…
araina-amd Apr 6, 2026
bc9a4e7
style: run black formatter on projection and pretrain_trainer
araina-amd Apr 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
198 changes: 135 additions & 63 deletions primus/backends/megatron/core/extensions/primus_turbo.py
View file
Open in desktop

Large diffs are not rendered by default.

720 changes: 0 additions & 720 deletions primus/backends/megatron/core/extensions/te_gemm_patch_wgrad.py
View file
Open in desktop

This file was deleted.

363 changes: 0 additions & 363 deletions primus/backends/megatron/core/extensions/te_group_gemm_patch_wgrad.py
View file
Open in desktop

This file was deleted.

Loading
Loading