ver217 232142f402 [utils] refactor profiler (#837) ...

* 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

2022-04-24 17:03:59 +08:00

..

legacy

[utils] refactor profiler (#837)

2022-04-24 17:03:59 +08:00

__init__.py

[utils] refactor profiler (#837)

2022-04-24 17:03:59 +08:00

extention.py

[utils] refactor profiler (#837)

2022-04-24 17:03:59 +08:00

profiler.py

[utils] refactor profiler (#837)

2022-04-24 17:03:59 +08:00

stateful_tensor_mem_extention.py

[utils] refactor profiler (#837)

2022-04-24 17:03:59 +08:00