Kai Wang (Victor Kai) 46931e3c32 [NFC] polish code colossalai/gemini/update/search_utils.py (#1557)

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 code colossalai/gemini/update/search_utils.py (#1557)

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

[hotfix] add deconstructor for stateful tensor (#848)

2022-04-24 15:03:04 +08:00

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

[hotfix] add deconstructor for stateful tensor (#848)

2022-04-24 15:03:04 +08:00

tensor_placement_policy.py

[gemini] accelerate adjust_layout() (#878)

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

tensor_utils.py

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00