/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/legacy/tensor/__init__.py
17 строк
418 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
from . import distspec from .compute_spec import ComputePattern, ComputeSpec from .dist_spec_mgr import DistSpecManager from .distspec import ReplicaSpec, ShardSpec from .process_group import ProcessGroup from .tensor_spec import ColoTensorSpec __all__ = [ "ComputePattern", "ComputeSpec", "distspec", "DistSpecManager", "ProcessGroup", "ColoTensorSpec", "ShardSpec", "ReplicaSpec", ]