ColossalAI/colossalai/gemini/chunk/__init__.py at 27ab5240965fc9cc0ec74ff48356abcbf098bd74 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 02:38:52 +00:00

Files

HELSON f69f9bf223 [zero] add chunk init function for users (#1729 )

* add chunk manager init function

* fix unit tests

* add comment

* add flush=True

2022-10-18 16:31:22 +08:00

5 lines

207 B

Python

Raw Blame History

 from .chunk import Chunk, ChunkFullError, TensorInfo, TensorState
 from .manager import ChunkManager
 from .search_utils import clasify_params, search_chunk_configuration
 from .utils import init_chunk_manager