diff --git a/aten/src/ATen/native/mps/operations/Convolution.mm b/aten/src/ATen/native/mps/operations/Convolution.mm index c047f8446b5c26..1d45be69a38795 100644 --- a/aten/src/ATen/native/mps/operations/Convolution.mm +++ b/aten/src/ATen/native/mps/operations/Convolution.mm @@ -700,8 +700,6 @@ Tensor _mps_convolution_transpose(const Tensor& input_t, IntArrayRef stride, IntArrayRef dilation, int64_t groups) { - TORCH_CHECK(input_t.dim() < 5, "ConvTranspose 3D is not supported on MPS"); - auto output_t = mps_convolution_transpose_forward(input_t, weight_t, padding, output_padding, stride, dilation, groups); return output_t;