ColossalAI/colossalai/device/__init__.py

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']