Add HuggingFace tp_plan support for AutoTP #7901
delock
merged
26 commits into
deepspeedai:master
from
delock:gma/autotp_improvement
Add HuggingFace tp_plan support for AutoTP
dfbc0f2a
remove llama test
fe30ae24
tests: fix tp_plan Qwen2 ZeRO2 setup
8f73781a
fix tests
a941292a
tests: align tp_plan e2e checks with autotp
a2aba588
add coverage of HF tp_plan
d48171da
tests: drop duplicate tp_plan priority case
d28922eb
tests: assert tp_plan source for custom model
a840f1f3
remove cosmetic change in engine.py
5db339bd
add comments for TP config priority
a3af0605
cleanup: remove dead resolve_tp_config and consolidate tests
fee78e6d
style: yapf formatting and remove unused import
1e08f6a2
simplify _get_hf_tp_plan: remove unused config-only path
db1b8f0d
add TODO for unsupported HF tp_plan partition types
1b3243aa
docs: add HuggingFace tp_plan documentation across tutorials, config,…
8870c987
chatgpt-codex-connector
commented
on 2026-03-13
fix: replace torch.distributed with deepspeed.comm in tp_plan tests
4b2f68b0
fix: _get_hf_tp_plan falls through to config when _tp_plan is None/empty
847d6e0b
docs: add PR link to tp_plan blog update
7e6038ae
Add info log when HuggingFace tp_plan is activated
a91aea56
fix: propagate full module path in tp_plan replacement and prefer bas…
f5bc5b68
test: add deep-model test for full_name propagation in _replace_module
9db02c90
Merge branch 'master' into gma/autotp_improvement
2facb27e
disabled auto-merge 86 days ago
Manually disabled by user
Manually disabled by user
fix: fall back to AutoTP when tp_plan contains unsupported styles
d83f0d3f
Merge remote-tracking branch 'delock/gma/autotp_improvement' into gma…
cabccf8d
Merge branch 'master' into gma/autotp_improvement
cbae7f19
tohtana
approved these changes
on 2026-03-23
Merge branch 'master' into gma/autotp_improvement
cad066dd
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub