ColossalAI/colossalai/gemini/update/__init__.py

from .chunkv2 import ChunkV2
from .chunk_mgrv2 import ChunkManagerV2
from .search_utils import clasify_params, search_chunk_configuration
[polish] add update directory in gemini; rename AgChunk to ChunkV2 (#1432) 2022-08-10 08:40:29 +00:00			`from .chunkv2 import ChunkV2`
[zero] add chunk_managerV2 for all-gather chunk (#1441) 2022-08-11 11:17:24 +00:00			`from .chunk_mgrv2 import ChunkManagerV2`
[zero] add chunk size searching algorithm for parameters in different groups (#1436) 2022-08-11 05:32:19 +00:00			`from .search_utils import clasify_params, search_chunk_configuration`