Support customized FSDP AGRS and AGRS Dispatcher#1554

Open

HIT-cwh wants to merge 9 commits intoInternLM:mainfrom

HIT-cwh:cwh/support_nvls_agrs

Commits on Jan 12, 2026

[Feature] support custom fsdp agrs for pt2.8

HIT-cwh
authored and
root
committed
[Feature] support custom fsdp agrs for pt2.6

HIT-cwh
authored and
root
committed
delete useless codes

HIT-cwh
authored and
root
committed
refactor

HIT-cwh
authored and
root
committed
move to op folder

HIT-cwh
authored and
root
committed
replace print with logger.info

HIT-cwh
authored and
root
committed

Commits on Jan 14, 2026

Commits on Mar 11, 2026

set num_buffers of AllGatherManager and ReduceScatterManager to 1
HIT-cwh
committed