Hongxin Liu 1b387ca9fe [shardformer] refactor pipeline grad ckpt config (#5646) ... * [shardformer] refactor pipeline grad ckpt config * [shardformer] refactor pipeline grad ckpt config * [pipeline] fix stage manager		2024-04-25 15:19:30 +08:00
..
schedule	[shardformer] fix chatglm implementation (#5644)	2024-04-25 14:41:17 +08:00
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
p2p.py	[pipeline] A more general _communicate in p2p (#5062)	2024-01-08 15:37:27 +08:00
stage_manager.py	[shardformer] refactor pipeline grad ckpt config (#5646)	2024-04-25 15:19:30 +08:00