github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero/gemini/chunk/__init__.py

7 lines
342 B
Python
Raw Normal View History

[zero] add chunk init function for users (#1729) * add chunk manager init function * fix unit tests * add comment * add flush=True
2022-10-18 08:31:22 +00:00
from .chunk import Chunk, ChunkFullError, TensorInfo, TensorState
from .manager import ChunkManager
[Gemini] NFC, polish search_chunk_configuration (#2107)
2022-12-09 07:00:39 +00:00
from .search_utils import classify_params_by_dp_degree, search_chunk_configuration
[zero] add chunk init function for users (#1729) * add chunk manager init function * fix unit tests * add comment * add flush=True
2022-10-18 08:31:22 +00:00
from .utils import init_chunk_manager
[Gemini] chunk init use OrderedParamGenerator (#2110)
2022-12-11 13:41:13 +00:00
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
__all__ = ["Chunk", "ChunkManager", "classify_params_by_dp_degree", "search_chunk_configuration", "init_chunk_manager"]