github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero/utils
History
ver217 dcca614eee
[hotfix] fix test_stateful_tensor_mgr (#762) 		2022-04-14 15:50:09 +08:00
..
__init__.py [zero] use factory pattern for tensor_placement_policy (#752) 2022-04-14 11:07:29 +08:00
stateful_tensor_mgr.py [zero] refactor memstats_collector (#746) 2022-04-14 12:01:12 +08:00
tensor_placement_policy.py [hotfix] fix test_stateful_tensor_mgr (#762) 2022-04-14 15:50:09 +08:00
zero_hook.py [zero] refactor memstats_collector (#746) 2022-04-14 12:01:12 +08:00