You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

LuGY a78daf6180 [shardformer] support interleaved pipeline (#4448) ... * support interleaved pipeline * fix unit test * remove virtual stage test in stage mgr * add droped type hint and updated bwd		1 year ago
..
__init__.py	[cluster] add process group mesh (#4039)	1 year ago
device_mesh_manager.py	[API] implement device mesh manager (#3221)	2 years ago
dist_coordinator.py	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
process_group_manager.py	[booster] implemented the cluster module (#3191)	2 years ago
process_group_mesh.py	[shardformer] support interleaved pipeline (#4448)	1 year ago