[Pytorch Profiler] Move start timestamp to end of start callback (#62191)
authorKimish Patel <kimishpatel@fb.com>
Sat, 14 Aug 2021 04:37:57 +0000 (21:37 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Sat, 14 Aug 2021 04:40:12 +0000 (21:40 -0700)
commit44b12ba8624ee9f0d5c77babc0d2c4b9012fc1b5
tree209fd654b55f0be5aa8eb7499044c01d2cf63f41
parent54f2eb6e7e313d84abfe7e1f3781998979732be2
[Pytorch Profiler] Move start timestamp to end of start callback (#62191)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62191

This moves start timestamping to end of callback. This way we dont
account for callstack/module hierarchy related overhead in op runtime.

Test Plan:
CI

Imported from OSS

Reviewed By: ilia-cher

Differential Revision: D29910519

fbshipit-source-id: f462031a81ae12b3db7993cf482e5ad93a35e096
torch/csrc/autograd/profiler_kineto.cpp