ColossalAI

Commit Graph

Author	SHA1	Message	Date
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	7e8114a8dd	[hotfix] skipped unsafe test cases (#1282 )	2022-07-13 00:08:59 +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	5581170890	[fx] fixed huggingface OPT and T5 results misalignment (#1227 )	2022-07-07 16:29:58 +08:00
Frank Lee	84f2298a96	[fx] added patches for tracing swin transformer (#1228 )	2022-07-07 15:20:13 +08:00
Frank Lee	37fcf96b7f	[fx] fixed timm tracing result misalignment (#1225 )	2022-07-07 14:45:15 +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	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

17 Commits (11d1436a6712ab113991f209219deb3d3529ce79)