pytorch
a25df29c - [ao] Updated ModelReport function calls to show not dependent on Fx GraphMode (#81252)

Commit
2 years ago
[ao] Updated ModelReport function calls to show not dependent on Fx GraphMode (#81252) Summary: Before, all the function calls for the ModelReport object were dependent on the Fx Graph Mode workflow. However, in reality, this was not true and the only requirement that was needed was for the model to be a traceable GraphModule. This also helped keep the ModelReport class as detached from the Fx Workflow as possible so that it can be used as a more all purpose tool in the future. This updated all the references to make sure that it wasn't specifically referencing that a Fx Graph Mode workflow is needed, and is instead more general since all we really need is a traceable model. Test Plan: python test/test_quantization.py TestFxModelReportClass Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/81252 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading