Handle non-contiguous inputs with mkldnn convolution.#16300
Conversation
facebook-github-bot
left a comment
There was a problem hiding this comment.
@gchanan has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
I wonder if there's a generic way to test for non-contiguous inputs. Do we do it currently for generically written tests in test_nn.py? |
|
@dzhulgakov we only test non contiguous inputs to NN modules, but it didn't actually matter in this case: there was already an explicit test for non-contiguous convolution weights (although not biases, I added a test for that), but it didn't exercise mkldnn because we only use mkldnn for float32 and we run autograd tests in float64... A test revamp would at least check float32 as well (and if it made sense, direct calls to different convolution algorithms). |
Summary: Fixes pytorch/pytorch#16018. Backwards appears to be fine because the derivative is written in terms of mkldnn_convolution itself. Pull Request resolved: pytorch/pytorch#16300 Differential Revision: D13797776 Pulled By: gchanan fbshipit-source-id: 68a990b8a3c186412a99d176931314806c9ed7bf
Summary: Fixes pytorch#16018. Backwards appears to be fine because the derivative is written in terms of mkldnn_convolution itself. Pull Request resolved: pytorch#16300 Differential Revision: D13797776 Pulled By: gchanan fbshipit-source-id: 68a990b8a3c186412a99d176931314806c9ed7bf
Fixes #16018.
Backwards appears to be fine because the derivative is written in terms of mkldnn_convolution itself.