ColossalAI/colossalai/gemini
ver217 846406a07a
[gemini] fix auto tensor placement policy (#775)
2022-04-16 21:29:31 +08:00
..
__init__.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
stateful_tensor_mgr.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
tensor_placement_policy.py [gemini] fix auto tensor placement policy (#775) 2022-04-16 21:29:31 +08:00