github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/utils/profiler/__init__.py

3 lines
82 B
Python
Raw Normal View History

Added Profiler Context to manage all profilers (#340)
2022-03-09 08:12:41 +00:00
from .comm_profiler import CommProfiler
from .prof_utils import ProfilerContext