pytorch
acd2f21e - [Profiler] Update python binding type annotations (#85722)

Commit
2 years ago
[Profiler] Update python binding type annotations (#85722) The annotations for `torch._C._profiler` have gotten a bit stale. This PR simply brings them up to date. There is one small quality of life change that alters behavior: instead of returning device type and index separately we return a `torch.device` object. Differential Revision: [D39852803](https://our.internmc.facebook.com/intern/diff/D39852803/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85722 Approved by: https://github.com/chaekit
Author
Taylor Robie
Committer
Parents
Loading