Taylor Robie
014a333df3
[Profiler][Minor] Extend Python bindings ( #83622 )
...
Adding some fields which are needed for memory profiling.
Differential Revision: [D38528382](https://our.internmc.facebook.com/intern/diff/D38528382/ )
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83622
Approved by: https://github.com/Gamrix
2022-08-26 20:03:24 +00:00
PyTorch MergeBot
d5af2a70ba
Revert "[TorchTidy] Adding support for unique tensor identifiers ( #80266 )"
...
This reverts commit b6ba41921dhttps://github.com/pytorch/pytorch/pull/80266 on behalf of https://github.com/malfet due to Broke number of trunk jobs, see b6ba41921d
2022-08-25 05:09:12 +00:00
John Clow
b6ba41921d
[TorchTidy] Adding support for unique tensor identifiers ( #80266 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80266
Approved by: https://github.com/robieta
2022-08-25 03:52:19 +00:00
Seonglyong Gong
fa241fd50e
[Profiler] record nn.Module's parameters ( #83209 )
...
Summary:
Record nn.Module's parameters for detaild memory profiling:
- extend 'module_' in value cache & NNModuleInfo to save parameters
- python binding and unit test case
Test Plan: buck run mode/opt //caffe2/test:profiler -- -r test_nnmodule
Differential Revision: D38379717
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83209
Approved by: https://github.com/robieta
2022-08-24 08:17:20 +00:00
Taylor Robie
1fa9a377d0
[Profiler] Start moving python bindings out of autograd ( #82584 )
...
A lot of profiler code still lives in autograd for historic reasons. However as we formalize and clean up profiler internals it makes sense to pull more and more into the profiler folders/namespace. For now I'm just moving some of the core config data structures and those related to `torch::profiler::impl::Result` to keep the scope manageable.
Differential Revision: [D37961462](https://our.internmc.facebook.com/intern/diff/D37961462/ )
**NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D37961462/ )!
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82584
Approved by: https://github.com/albanD , https://github.com/Gamrix
2022-08-19 17:15:18 +00:00
