4 Commits

Author	SHA1	Message	Date
Frank Lee	5a52e21fe3	[test] fixed the activation codegen test (#1447) ... * [test] fixed the activation codegen test * polish code	2022-08-12 14:52:31 +08:00
Boyuan Yao	5774fe0270	[fx] Use colossalai checkpoint and add offload recognition in codegen (#1439) ... * [fx] Use colossalai.utils.checkpoint to replace torch.utils.checkpoint for offload activation and add offload annotation recognition in codegen * [fx] Use colossalai.utils.checkpoint to replace torch.utils.checkpoint for offload activation and add offload annotation recognition in codegen * Modification of test and add TODO in codegen * [fx] Modification of colossal ckpt usage * [fx] add gpc.destroy() to test_codegen	2022-08-12 12:23:30 +08:00
Frank Lee	cd063ac37f	[fx] added activation checkpoint codegen support for torch < 1.12 (#1359)	2022-07-25 23:35:31 +08:00
Frank Lee	644582eee9	[fx] added activation checkpoint codegen (#1355)	2022-07-25 09:39:10 +08:00