ColossalAI/colossalai/utils/profiler/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .comm_profiler import CommProfiler
from .pcie_profiler import PcieProfiler
from .prof_utils import ProfilerContext
Added Profiler Context to manage all profilers (#340) 2022-03-09 08:12:41 +00:00			`from .comm_profiler import CommProfiler`
Added PCIE profiler to dectect data transmission (#373) 2022-03-10 08:24:57 +00:00			`from .pcie_profiler import PcieProfiler`
Added Profiler Context to manage all profilers (#340) 2022-03-09 08:12:41 +00:00			`from .prof_utils import ProfilerContext`