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] 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_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

stateful_tensor.py

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

2022-04-24 15:03:04 +08:00

tensor_placement_policy.py

[gemini] accelerate adjust_layout() (#878)

2022-04-26 18:08:31 +08:00

tensor_utils.py

[gemini] add GeminiMemoryManger (#832)

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