pytorch
e77f28a0 - [Quant] Add fused ConvAddReLU2d module for onednn backend (#91154)

Commit
1 year ago
[Quant] Add fused ConvAddReLU2d module for onednn backend (#91154) **Summary** Post op fusion can reduce data movement overhead and improve inference performance. This PR adds fused ConvAddReLU2d module for onednn backend, which will be used for int8 inference with onednn backend. Cannot call this module with other quantization backends otherwise an error is thrown. **Test plan** ``` python -m pytest test_quantization.py -k test_conv2d_add_relu ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/91154 Approved by: https://github.com/jgong5, https://github.com/jerryzh168
Committer
Parents
Loading