| .. |
|
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.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
|
[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.py
|
[hotfix] add deconstructor for stateful tensor (#848)
|
2022-04-24 15:03:04 +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 |
|
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 |
9823cbf24b