ColossalAI/colossalai/zero/utils
ver217 be01db37c8
[tensor] refactor chunk mgr and impl MemStatsCollectorV2 (#1077)
* polish chunk manager

* polish unit test

* impl add_extern_static_tensor for chunk mgr

* add mem stats collector v2

* polish code

* polish unit test

* polish code

* polish get chunks
2022-06-09 20:56:34 +08:00
..
__init__.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
zero_hook.py [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
zero_hook_v2.py [tensor] refactor chunk mgr and impl MemStatsCollectorV2 (#1077) 2022-06-09 20:56:34 +08:00