/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/fx/profiler/__init__.py
27 строк
768 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
from .._compatibility import is_compatible_with_meta if is_compatible_with_meta(): from .opcount import flop_mapping from .profiler import profile_function, profile_method, profile_module from .shard_utils import ( calculate_bwd_time, calculate_fwd_in, calculate_fwd_out, calculate_fwd_time, calculate_fwd_tmp, ) from .tensor import MetaTensor else: from .experimental import ( meta_profiler_function, meta_profiler_module, profile_function, profile_method, profile_module, calculate_fwd_in, calculate_fwd_tmp, calculate_fwd_out, ) from .dataflow import GraphInfo from .memory_utils import activation_size, is_inplace, parameter_size