ColossalAI/colossalai/device/__init__.py

5 lines
139 B
Python

from .alpha_beta_profiler import AlphaBetaProfiler
from .calc_pipeline_strategy import alpa_dp
__all__ = ['AlphaBetaProfiler', 'alpa_dp']