pytorch
9ed49449 - [SR] Add net level record functions (#68091)

Commit
3 years ago
[SR] Add net level record functions (#68091) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68091 Add record functions for recording perf stats on the entire network. Note that this is backed by the same pre-sampling mechanism as the op record functions, so net level stats get logged relatively infrequently. (If this is not acceptable, we can not use pre-sampling at the cost of a little bit of perf, every inference will require an RNG call) Reviewed By: hlu1 Differential Revision: D32296756 fbshipit-source-id: 09ff16c942f3bfc8f4435d6cca2be4a6b8dc6091
Author
Mike Iovine
Parents
Loading