Use MIOpen for transpose convolutions (#26172)

lcskrishna · facebook-github-bot · commit e86d99ae8803 · 2019-09-14T23:23:53.000-07:00
Summary: Provides significant performance uplift where used. Pull Request resolved: #26172 Differential Revision: D17374862 Pulled By: bddppq fbshipit-source-id: 85d2df3c67b8935bc54f3a81a912a25c0102743a
diff --git a/aten/src/ATen/native/Convolution.cpp b/aten/src/ATen/native/Convolution.cpp
@@ -143,7 +143,6 @@ auto ConvParams::use_miopen(const at::Tensor& input) const -> bool {
          && input.is_cuda()
          && input.dim() <= MIOPEN_DIM_MAX
          && !(groups > 1 && is_dilated()) // MIOpen currently does not support dilation with groups of size > 1
-         && !transposed
          ;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,6 @@ auto ConvParams::use_miopen(const at::Tensor& input) const -> bool {`
`143`	`143`	`&& input.is_cuda()`
`144`	`144`	`&& input.dim() <= MIOPEN_DIM_MAX`
`145`	`145`	`&& !(groups > 1 && is_dilated()) // MIOpen currently does not support dilation with groups of size > 1`
`146`		`- && !transposed`
`147`	`146`	`;`
`148`	`147`	`}`
`149`	`148`