pytorch
a26238da - Enable using `torch.autograd.profiler.record_function` as decorator (#30861)

Commit
5 years ago
Enable using `torch.autograd.profiler.record_function` as decorator (#30861) Summary: ```python record_function('my_func') def f(x, y): return x + y with profile() as p: f(1, 2) print(prof.key_averages().table()) ``` ``` ------------------------------------ --------------- --------------- --------------- --------------- --------------- --------------- Name Self CPU total % Self CPU total CPU total % CPU total CPU time avg Number of Calls ------------------------------------ --------------- --------------- --------------- --------------- --------------- --------------- my_func 85.42% 86.796us 87.27% 88.670us 88.670us 1 ------------------------------------ --------------- --------------- --------------- --------------- --------------- --------------- Self CPU time total: 101.606us ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/30861 Differential Revision: D18857993 Pulled By: bddppq fbshipit-source-id: eb6b8e2a8d4f3a7f8e5b4cb3da1ee3320acb1ae7
Author
Parents
Loading