28 Commits (a7cda6f57dd8f7fc0d5c7ae3b4b5cf9cd8140c13)

Author	SHA1	Message	Date
Xue Fuzhao	e070ca45c6	[NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563)	2022-09-08 22:11:04 +08:00
Super Daniel	d8a5aded19	[hotfix] change namespace for meta_trace. (#1541)	2022-09-06 11:46:12 +08:00
Super Daniel	70129603aa	[fx] support meta tracing for aten level computation graphs like functorch. (#1536) ... * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] remove redundant import. * [fx] add docstring.	2022-09-05 12:10:09 +08:00
YuliangLiu0306	4b3d6caeb3	[fx]patch nn.functional convolution (#1528)	2022-09-01 19:05:07 +08:00
YuliangLiu0306	413c053453	[autoparallel] add cost graph class (#1481) ... * [autoparallel] add cost graph class * polish code	2022-08-25 17:19:59 +08:00
Frank Lee	3da68d6b1b	[fx] fixed adapative pooling size concatenation error (#1489)	2022-08-25 09:05:07 +08:00
Frank Lee	7d6293927f	[fx] patched torch.max and data movement operator (#1391) ... * [fx] patched torch.max and data movement operator * polish code	2022-08-01 15:31:50 +08:00
Frank Lee	ad678921db	[fx] patched torch.full for huggingface opt (#1386)	2022-07-29 17:56:28 +08:00
Super Daniel	be229217ce	[fx] add torchaudio test (#1369) ... * [fx]add torchaudio test * [fx]add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test and test patches * Delete ~ * [fx] add patches and patches test * [fx] add patches and patches test * [fx] fix patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] merge upstream * [fx] fix import errors	2022-07-27 11:03:14 +08:00
ver217	d068af81a3	[doc] update rst and docstring (#1351) ... * update rst * add zero docstr * fix docstr * remove fx.tracer.meta_patch * fix docstr * fix docstr * update fx rst * fix fx docstr * remove useless rst	2022-07-21 15:54:53 +08:00
Frank Lee	274c1a3b5f	[fx] fixed apex normalization patch exception (#1352)	2022-07-21 15:29:11 +08:00
Frank Lee	05fae1fd56	[fx] added activation checkpointing annotation (#1349) ... * [fx] added activation checkpointing annotation * polish code * polish code	2022-07-21 11:14:28 +08:00
YuliangLiu0306	942c8cd1fb	[fx] refactor tracer to trace complete graph (#1342) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * [fx] refactor tracer to trace complete graph * add comments and solve conflicts.	2022-07-20 11:20:38 +08:00
YuliangLiu0306	4631fef8a0	[fx]refactor tracer (#1335)	2022-07-19 15:50:42 +08:00
Frank Lee	75abc75c15	[fx] fixed compatiblity issue with torch 1.10 (#1331)	2022-07-18 11:41:27 +08:00
Frank Lee	b2475d8c5c	[fx] fixed unit tests for torch 1.12 (#1327)	2022-07-15 18:22:15 +08:00
Frank Lee	4f4d8c3656	[fx] added apex normalization to patched modules (#1300) ... * [fx] added apex normalization to patched modules * remove unused imports	2022-07-14 14:24:13 +08:00
Frank Lee	4a09fc0947	[fx] fixed tracing with apex-based T5 model (#1252) ... * [fx] fixed tracing with apex-based T5 model * polish code * polish code	2022-07-12 15:19:25 +08:00
Frank Lee	7531c6271f	[fx] refactored the file structure of patched function and module (#1238) ... * [fx] refactored the file structure of patched function and module * polish code	2022-07-12 15:01:58 +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
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
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
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