ColossalAI/colossalai/device/__init__.py

from .alpha_beta_profiler import AlphaBetaProfiler
from .calc_pipeline_strategy import alpa_dp

__all__ = ["AlphaBetaProfiler", "alpa_dp"]
[device] alpha beta profiler (#2311) * [device] alpha beta profiler * add usage * fix variable name 2023-01-05 08:39:55 +00:00			`from .alpha_beta_profiler import AlphaBetaProfiler`
[autoparallel] Add alpha beta (#1973) * Add alpha beta * Fix test * Fix test 2022-11-17 08:01:14 +00:00			`from .calc_pipeline_strategy import alpa_dp`

[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format 2023-09-19 06:20:26 +00:00			`__all__ = ["AlphaBetaProfiler", "alpa_dp"]`