pytorch
232fbd90 - [TorchDynamo]: fused bias for cpu convolution path (#87050)

Commit
2 years ago
[TorchDynamo]: fused bias for cpu convolution path (#87050) For aten.convolution CPU path, the bias always can be fused, so this PR adds a device check: if inputs' device is CPU, we will fuse it for a good performance. Pull Request resolved: https://github.com/pytorch/pytorch/pull/87050 Approved by: https://github.com/jgong5, https://github.com/jansel
Author
Committer
Parents
Loading