pytorch
6fe234ec - pnp: move shadow loggers to parent module (#91428)

Commit
1 year ago
pnp: move shadow loggers to parent module (#91428) Summary: Before this PR, PNP added shadow loggers to insides of the shadow wrapper modules. This PR moves those loggers to the parent module. There are a couple of benefits: 1. this will unbreak features of quantization API which don't support loggers (such as hardcoding model output to be quantized) 2. this makes it easier to look at the parent graph and visualize what is logged, since now all the logging is in the same graph 3. this will make it easier to implement features such as propagation error calculation in the future Test plan: ``` python test/test_quantization.py -k NShadows ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/91428 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading