AstrAI/astrai/parallel
ViperEkura b092316385 feat : add distributed checkpoint via executor checkpoint_context
- Add checkpoint_context context manager to BaseExecutor with entry/exit barrier
- Add _gather_state_dict hook overridden per executor (template method)
- DDPExecutor skips unwrap on non-rank-0 to avoid redundant state_dict gather
- FSDPExecutor uses rank0_only=True to reduce memory on non-writers
- Remove redundant rank-0 guard from Checkpoint.save and manual barrier from Callback
2026-07-13 12:27:09 +08:00
..
__init__.py feat: 新增FSDP并行后端 2026-05-25 19:43:14 +08:00
executor.py feat : add distributed checkpoint via executor checkpoint_context 2026-07-13 12:27:09 +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