ColossalAI/colossalai/gemini
ver217 0dea140760
[hotfix] add deconstructor for stateful tensor (#848)
* add deconstructor for stateful tensor

* fix colo init context
2022-04-24 15:03:04 +08:00
..
memory_tracer [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +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 [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
tensor_utils.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00