Jiarui Fang 0e199d71e8 [hotfix] fx get comm size bugs (#1233) ... * init a checkpoint dir * [checkpoint]support resume for cosinewarmuplr * [checkpoint]add unit test * fix some bugs but still not OK * fix bugs * make it faster * [checkpoint]support generalized scheduler * polish * [tensor] torch function return colotensor * polish * fix bugs * remove debug info * polish * polish * [tensor] test_model pass unittests * polish * [hotfix] fx get comm size bug Co-authored-by: ZhaoYi1222 <zhaoyi9499@gmail.com>		2022-07-08 10:54:41 +08:00
..
__init__.py	[fx]add autoparallel passes (#1121)	2022-06-15 16:36:46 +08:00
adding_split_node_pass.py	[fx]add uniform policy (#1208)	2022-07-06 13:48:11 +08:00
meta_info_prop.py	[fx]get communication size between partitions (#1224)	2022-07-07 16:22:00 +08:00
shard_1d_pass.py	[hotfix] fx shard 1d pass bug fixing (#1220)	2022-07-07 13:37:31 +08:00
utils.py	[hotfix] fx get comm size bugs (#1233)	2022-07-08 10:54:41 +08:00