ColossalAI/colossalai/utils/profiler
Jie Zhu 73d36618a6
[profiler] add MemProfiler (#356)
* 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
2022-03-29 12:48:34 +08:00
..
__init__.py [profiler] add MemProfiler (#356) 2022-03-29 12:48:34 +08:00
comm_profiler.py fixed error when no collective communication in CommProfiler 2022-03-14 17:21:00 +08:00
mem_profiler.py [profiler] add MemProfiler (#356) 2022-03-29 12:48:34 +08:00
pcie_profiler.py fixed error when no collective communication in CommProfiler 2022-03-14 17:21:00 +08:00
prof_utils.py polished output format for communication profiler and pcie profiler (#404) 2022-03-14 16:07:45 +08:00