mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
415 B
16 lines
415 B
from .colo_module import ColoModule
|
|
from .linear import ColoLinear
|
|
from .embedding import ColoEmbedding
|
|
from .module_utils import register_colo_module, is_colo_module, get_colo_module, init_colo_module, check_colo_module
|
|
|
|
__all__ = [
|
|
'ColoModule',
|
|
'register_colo_module',
|
|
'is_colo_module',
|
|
'get_colo_module',
|
|
'init_colo_module',
|
|
'check_colo_module',
|
|
'ColoLinear',
|
|
'ColoEmbedding',
|
|
]
|