Zangwei Zheng 9823cbf24b [NFC] polish colossalai/gemini/update/chunkv2.py code style (#1565)

2022-09-08 22:11:04 +08:00

..

memory_tracer

[refactor] move chunk and chunkmgr to directory gemini (#1182)

2022-06-29 13:31:02 +08:00

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

[refactor] move chunk and chunkmgr to directory gemini (#1182)

2022-06-29 13:31:02 +08:00

chunk_mgr.py

[zero] chunk manager allows filtering ex-large params (#1393)

2022-08-02 10:40:27 +08:00

chunk.py

[hotfix] fix a running error in test_colo_checkpoint.py (#1387)

2022-07-29 15:58:06 +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

make AutoPlacementPolicy configurable (#1191)

2022-06-30 15:18:30 +08:00

stateful_tensor_container.py

[gemini] add stateful tensor container (#867)

2022-04-25 14:58:16 +08:00

stateful_tensor_mgr.py

[gemini] accelerate adjust_layout() (#878)

2022-04-26 18:08:31 +08:00

stateful_tensor.py

…

tensor_placement_policy.py

[gemini] accelerate adjust_layout() (#878)

2022-04-26 18:08:31 +08:00

tensor_utils.py

…