Zangwei Zheng 9823cbf24b [NFC] polish colossalai/gemini/update/chunkv2.py code style (#1565)		2022-09-08 22:11:04 +08:00
..
memory_tracer	…
ophooks	[hotfix] remove potiential circle import (#1307)	2022-07-14 13:44:26 +08:00
paramhooks	[hotfix] remove potiential circle import (#1307)	2022-07-14 13:44:26 +08:00
update	[NFC] polish colossalai/gemini/update/chunkv2.py code style (#1565)	2022-09-08 22:11:04 +08:00
__init__.py	…
chunk.py	[hotfix] fix a running error in test_colo_checkpoint.py (#1387)	2022-07-29 15:58:06 +08:00
chunk_mgr.py	[zero] chunk manager allows filtering ex-large params (#1393)	2022-08-02 10:40:27 +08:00
gemini_context.py	…
gemini_mgr.py	[doc] update rst and docstring (#1351)	2022-07-21 15:54:53 +08:00
placement_policy.py	…
stateful_tensor.py	…
stateful_tensor_container.py	…
stateful_tensor_mgr.py	…
tensor_placement_policy.py	…
tensor_utils.py	…