Files
ColossalAI/colossalai/device/__init__.py
Genghan Zhang 6630d45546 [autoparallel] Add alpha beta (#1973)
* Add alpha beta

* Fix test

* Fix test
2022-11-17 16:01:14 +08:00

5 lines
140 B
Python

from .calc_pipeline_strategy import alpa_dp
from .profile_alpha_beta import profile_alpha_beta
__all__ = ['profile_alpha_beta', 'alpa_dp']