Commit Graph

8 Commits (7a58dc5ad244f68d162e34a269a4a7c96f9896ab)

Author SHA1 Message Date
Boyuan Yao 47fd8e4a02
[utils] Add use_reetrant=False in utils.activation_checkpoint (#1460)
2 years ago
Jiarui Fang 3b500984b1
[tensor] fix some unittests (#1234)
2 years ago
HELSON e5d615aeee
[hotfix] fix bugs in testing (#659)
3 years ago
Jiarui Fang 21dc54e019
[zero] memtracer to record cuda memory usage of model data and overall system (#395)
3 years ago
Frank Lee 1e4bf85cdb fixed bug in activation checkpointing test (#387)
3 years ago
LuGY de46450461 Added activation offload (#331)
3 years ago
Frank Lee cd9c28e055
added CI for unit testing (#69)
3 years ago
zbian 404ecbdcc6 Migrated project
3 years ago