pytorch
ea86eb31 - inductor: fallback ConvTranspose when output_padding is big (#100846)

Commit
2 years ago
inductor: fallback ConvTranspose when output_padding is big (#100846) Fixes https://github.com/pytorch/pytorch/issues/100225. Do not use mkldnn when `output_padding` is big to align the behavior with eager mode: https://github.com/pytorch/pytorch/blob/7d0e4e2aa843ef5d73646f5b304914d2b65db93c/aten/src/ATen/native/Convolution.cpp#L500-L507 Pull Request resolved: https://github.com/pytorch/pytorch/pull/100846 Approved by: https://github.com/jgong5, https://github.com/EikanWang, https://github.com/jansel
Author
Committer
Parents
Loading