ColossalAI/colossalai/gemini
HELSON 3107817172
[gemini] add stateful tensor container (#867)
2022-04-25 14:58:16 +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_container.py [gemini] add stateful tensor container (#867) 2022-04-25 14:58:16 +08:00
stateful_tensor_mgr.py [gemini] polish code (#855) 2022-04-25 10:40:14 +08:00
tensor_placement_policy.py [gemini] polish code (#855) 2022-04-25 10:40:14 +08:00
tensor_utils.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00