From 3f843fe51872a002e90c38cd699033052b347c5e Mon Sep 17 00:00:00 2001
From: Max-Heinrich Laves <laves@imfusion.com>
Date: Wed, 22 Jan 2025 10:58:46 +0100
Subject: [PATCH] removed check for ConvTranspose3D on MPS

---
 aten/src/ATen/native/mps/operations/Convolution.mm | 2 --
 1 file changed, 2 deletions(-)

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;