mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-15 23:23:11 +00:00
73d36618a6
* add memory trainer hook * fix bug * add memory trainer hook * fix import bug * fix import bug * add trainer hook * fix #370 git log bug * modify `to_tensorboard` function to support better output * remove useless output * change the name of `MemProfiler` * complete memory profiler * replace error with warning * finish trainer hook * modify interface of MemProfiler * modify `__init__.py` in profiler * remove unnecessary pass statement * add usage to doc string * add usage to trainer hook * new location to store temp data file
6 lines
270 B
Python
6 lines
270 B
Python
from .comm_profiler import CommProfiler
|
|
from .pcie_profiler import PcieProfiler
|
|
from .prof_utils import ProfilerContext, BaseProfiler
|
|
from .mem_profiler import MemProfiler
|
|
|
|
__all__ = ['BaseProfiler', 'CommProfiler', 'PcieProfiler', 'MemProfiler', 'ProfilerContext'] |