mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-24 18:51:33 +00:00
554aa9592e
* [legacy] move communication to legacy (#4640) * [legacy] refactor logger and clean up legacy codes (#4654) * [legacy] make logger independent to gpc * [legacy] make optim independent to registry * [legacy] move test engine to legacy * [legacy] move nn to legacy (#4656) * [legacy] move nn to legacy * [checkpointio] fix save hf config * [test] remove useledd rpc pp test * [legacy] fix nn init * [example] skip tutorial hybriad parallel example * [devops] test doc check * [devops] test doc check
22 lines
913 B
Python
22 lines
913 B
Python
from .cache_embedding import (
|
|
CachedEmbeddingBag,
|
|
CachedParamMgr,
|
|
EvictionStrategy,
|
|
LimitBuffIndexCopyer,
|
|
ParallelCachedEmbeddingBag,
|
|
ParallelCachedEmbeddingBagTablewise,
|
|
ParallelCachedEmbeddingBagTablewiseSpiltCache,
|
|
TablewiseEmbeddingBagConfig,
|
|
)
|
|
from .colo_module import ColoModule
|
|
from .embedding import ColoEmbedding
|
|
from .linear import ColoLinear
|
|
from .module_utils import check_colo_module, get_colo_module, init_colo_module, is_colo_module, register_colo_module
|
|
|
|
__all__ = [
|
|
'ColoModule', 'register_colo_module', 'is_colo_module', 'get_colo_module', 'init_colo_module', 'check_colo_module',
|
|
'ColoLinear', 'ColoEmbedding', 'CachedEmbeddingBag', 'ParallelCachedEmbeddingBag', 'CachedParamMgr',
|
|
'LimitBuffIndexCopyer', 'EvictionStrategy', 'ParallelCachedEmbeddingBagTablewise', 'TablewiseEmbeddingBagConfig',
|
|
'ParallelCachedEmbeddingBagTablewiseSpiltCache'
|
|
]
|