ViperEkura 4e38183e86 fix: make FSDP2 executor work with ABC+Generic model hierarchy ...

- Wrap each child module individually, skip root (CPython layout
  conflict between ABC+Generic and FSDP2 __class__ assignment)
- Remove manual unshard in clip_grad_norm (DTensor compatible)
- Fix _no_sync to iterate modules() instead of checking root
- Add reshard after unwrap_model
- Guard __init_subclass__ type resolution against dynamic subclasses
- Add fsdp2 to --parallel_mode CLI choices

2026-07-23 16:11:02 +08:00

..

__init__.py

feat: add FSDP2Executor using torch.distributed.fsdp.fully_shard API

2026-07-20 01:46:25 +08:00

executor.py

fix: make FSDP2 executor work with ABC+Generic model hierarchy

2026-07-23 16:11:02 +08:00

module.py

refactor: 优化参数传递，清理导入样式

2026-04-03 22:06:32 +08:00

setup.py

fix: auto-assign free port in spawn_parallel_fn to avoid EADDRINUSE

2026-07-12 19:21:56 +08:00