ColossalAI/colossalai/utils/profiler/__init__.py

2 lines
79 B
Python

from .comm_profiler import enable_communication_prof, communication_prof_show