github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 20:40:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

[misc] update pre-commit and run all files (#4752)

Browse Source 
* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format
This commit is contained in:
Hongxin Liu
2023-09-19 14:20:26 +08:00
committed by GitHub
parent 3c6b831c26
commit 079bf3cb26
1268 changed files with 50037 additions and 38444 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
colossalai/legacy/nn/parallel/layers/__init__.py
View File

@@ -14,8 +14,20 @@ 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'
"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",
]