You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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		2 years ago
..
codegen	[fx] Use colossalai checkpoint and add offload recognition in codegen (#1439)	2 years ago
passes	[fx] fix the false interpretation of algorithm 3 in https://arxiv.org/abs/1604.06174. (#1446)	2 years ago
tracer	…
__init__.py	…
proxy.py	…