/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/tensor/__init__.py
18 строк
600 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
from .colo_parameter import ColoParameter from .colo_tensor import ColoTensor from .comm_spec import CollectiveCommPattern, CommSpec from .param_op_hook import ColoParamOpHook, ColoParamOpHookManager from .utils import convert_dim_partition_dict, convert_parameter, merge_same_dim_mesh_list, named_params_with_colotensor __all__ = [ "ColoTensor", "convert_parameter", "named_params_with_colotensor", "ColoParameter", "ColoParamOpHook", "ColoParamOpHookManager", "CommSpec", "CollectiveCommPattern", "convert_dim_partition_dict", "merge_same_dim_mesh_list", ]