mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-17 00:41:39 +00:00
232142f402
* add model data profiler * add a subclass of torch.profiler.profile * refactor folder structure * remove redundant codes * polish code * use GeminiMemoryManager * fix import path * fix stm profiler ext * polish comments * remove useless file
7 lines
272 B
Python
7 lines
272 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']
|