pytorch/test/cpp
Kimish Patel 38c185189c [Pytorch Edge] Enable kineto profiler on mobile via EdgeKinetoProfiler (#62419)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62419

This diff adds support for cpu only kineto profiler on mobile. Thus
enabling chrome trace generation on mobile. This bring cpp API for
mobile profiling on part with Torchscript.
This is done via:
1. Utilizating debug handle annotations in KinetoEvent.
2. Adding post processing capability, via callbacks, to
KinetoThreadLocalState
3. Creating new RAII stype profiler, KinetoEdgeCPUProfiler, which can be
used in surrounding scope of model execution. This will write chrome
trace to the location specified in profiler constructor.

Test Plan:
MobileProfiler.ModuleHierarchy

Imported from OSS

Reviewed By: raziel

Differential Revision: D29993660

fbshipit-source-id: 0b44f52f9e9c5f5aff81ebbd9273c254c3c03299
2021-08-13 21:40:19 -07:00
..
api Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
c10d [Pytorch Profiler] Add support for adding module hierarchy to (#61792) 2021-08-13 21:39:10 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [Pytorch Edge] Enable kineto profiler on mobile via EdgeKinetoProfiler (#62419) 2021-08-13 21:40:19 -07:00
lite_interpreter_runtime [Pytorch Edge] Enable kineto profiler on mobile via EdgeKinetoProfiler (#62419) 2021-08-13 21:40:19 -07:00
rpc Remove process_group_agent and faulty_process_group_agent files (#62985) 2021-08-10 15:57:39 -07:00
tensorexpr [nnc] Removed const from all fields in IR. (#62336) 2021-08-03 11:44:36 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00