8 Commits (51f1ec96b0140f617749a48226d092018ca22434)

Author	SHA1	Message	Date
Frank Lee	2412429d54	[util] fixed activation checkpointing on torch 1.9 (#719)	3 years ago
LuGY	140263a394	[hotfix]fixed bugs of assigning grad states to non leaf nodes (#711) ... * fixed bugs of assigning grad states to non leaf nodes * use detach()	3 years ago
HELSON	e5d615aeee	[hotfix] fix bugs in testing (#659) ... * remove hybrid adam in test_moe_zero_optim * fix activation checkpointing and its unitest	3 years ago
LuGY	1e2557e801	[zero] fixed the activation offload (#647) ... * fixed the activation offload * polish	3 years ago
Liang Bowen	ec5086c49c	Refactored docstring to google style	3 years ago
LuGY	de46450461	Added activation offload (#331) ... * Added activation offload * Fixed the import bug, used the pytest	3 years ago
HELSON	0f8c7f9804	Fixed docstring in colossalai (#171)	3 years ago
zbian	404ecbdcc6	Migrated project	3 years ago