ViperEkura bcaa2d1ae0 fix: FSDP unwrap_model collective op and None guard ...

- unshard() and full_tensor() are collective ops, all ranks must participate
- Old code returned None on non-rank-0 before calling unshard, causing deadlock
- Fix: all ranks unshard/full_tensor, only rank-0 keeps the result
- Move create_ref_model to parallel/utils.py, accept executor+model directly
- Guard create_ref_model and sync_old_model against None on non-rank-0

2026-07-29 23:41:10 +08:00

..

__init__.py

fix: FSDP unwrap_model collective op and None guard

2026-07-29 23:41:10 +08:00

executor.py

fix: FSDP unwrap_model collective op and None guard

2026-07-29 23:41:10 +08:00

module.py

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

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

setup.py

refactor: move signal_handler from parallel/ to top-level for broader reuse

2026-07-28 10:36:17 +08:00

utils.py

fix: FSDP unwrap_model collective op and None guard

2026-07-29 23:41:10 +08:00