ColossalAI/colossalai/gemini
ver217 d7e0303d1e
[zero] use GeminiMemoryManager when sampling model data (#850)
2022-04-24 17:17:22 +08:00
..
memory_tracer [zero] use GeminiMemoryManager when sampling model data (#850) 2022-04-24 17:17:22 +08:00
__init__.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
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_mgr.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
tensor_placement_policy.py [zero] use GeminiMemoryManager when sampling model data (#850) 2022-04-24 17:17:22 +08:00
tensor_utils.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00