ColossalAI/colossalai/gemini
ver217 c4d903e64a
[gemini] accelerate adjust_layout() (#878)
* add lru cache

* polish code

* update unit test

* fix sharded optim
2022-04-26 18:08:31 +08:00
..
memory_tracer [zero] use GeminiMemoryManager when sampling model data (#850) 2022-04-24 17:17:22 +08:00
__init__.py
gemini_context.py [hotfix] add deconstructor for stateful tensor (#848) 2022-04-24 15:03:04 +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