119 Commits (7c7921f71bf93e739b1939c724a4cfe9cd405247)

Author	SHA1	Message	Date
Jiarui Fang	9bcd2fd4af	[tensor] a shorter shard and replicate spec (#1245)	2022-07-11 15:51:48 +08:00
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
YuliangLiu0306	2b7dca44b5	[fx]get communication size between partitions (#1224) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * [fx]get communication size between partitions. * polish	2022-07-07 16:22:00 +08:00
Frank Lee	84f2298a96	[fx] added patches for tracing swin transformer (#1228)	2022-07-07 15:20:13 +08:00
Frank Lee	b6cb5a47ad	[fx] added timm model tracing testing (#1221)	2022-07-07 14:02:17 +08:00
Jiarui Fang	db1bef9032	[hotfix] fx shard 1d pass bug fixing (#1220)	2022-07-07 13:37:31 +08:00
Frank Lee	11973d892d	[fx] added torchvision model tracing testing (#1216) ... * [fx] added torchvision model tracing testing * remove unused imports	2022-07-06 21:37:56 +08:00
XYE	291e22aac6	[fx] temporarily used (#1215)	2022-07-06 17:19:26 +08:00
Frank Lee	5da87ce35d	[fx] added testing for all albert variants (#1211)	2022-07-06 15:11:08 +08:00
Frank Lee	2d13a45a3b	[fx] added testing for all gpt variants (#1210) ... * [fx] added testing for all gpt variants * polish code * polish code	2022-07-06 14:03:13 +08:00
YuliangLiu0306	189946c5c4	[fx]add uniform policy (#1208) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * [fx]add uniform policy	2022-07-06 13:48:11 +08:00
Frank Lee	426a279ce7	[fx] added testing for all bert variants (#1207) ... * [fx] added testing for all bert variants * polish code	2022-07-06 10:50:49 +08:00
Frank Lee	f7878f465c	[fx] supported model tracing for huggingface bert (#1201) ... * [fx] supported model tracing for huggingface bert * polish test	2022-07-05 13:19:57 +08:00
Frank Lee	abf6a262dc	[fx] added module patch for pooling layers (#1197)	2022-07-04 15:21:26 +08:00
Frank Lee	2c8c05675d	[fx] patched conv and normalization (#1188)	2022-06-29 18:58:38 +08:00
Frank Lee	6d86f1bc91	[fx] supported data-dependent control flow in model tracing (#1185) ... * [fx] supported data-dependent control flow in model tracing * polish code	2022-06-29 15:05:25 +08:00
YuliangLiu0306	fcf55777dd	[fx]add autoparallel passes (#1121) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * feature/add autoparallel passes	2022-06-15 16:36:46 +08:00
Frank Lee	16302a5359	[fx] added unit test for coloproxy (#1119) ... * [fx] added unit test for coloproxy * polish code * polish code	2022-06-15 15:27:51 +08:00
Frank Lee	e1620ddac2	[fx] added coloproxy (#1115)	2022-06-15 10:47:57 +08:00