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_mgr.py

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

stateful_tensor.py

[hotfix] add deconstructor for stateful tensor (#848)

2022-04-24 15:03:04 +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