pytorch
ddedeab6 - ns for fx: bug fix for shadowing fp16 emulation patterns (#57024)

Commit
3 years ago
ns for fx: bug fix for shadowing fp16 emulation patterns (#57024) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57024 Enables shadow copies of fp16 emulation patterns where weights are cast to fp16 before being passed to linear. This previously did not work because copying of `call_method` nodes was not implemented. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_linear_fp16_vs_linear_fp16_shadow_activations ``` Reviewed By: jerryzh168 Differential Revision: D28030096 Pulled By: vkuzo fbshipit-source-id: 13a39ea6c106180df6d750246672286b58b4d04c
Author
Parents
Loading