From 69b511c2c738156bbb5c96619e1d1e5a7466aa94 Mon Sep 17 00:00:00 2001
From: Ronian526 <ziyu_zhang@apple.com>
Date: Tue, 31 Jan 2023 14:54:43 -0800
Subject: [PATCH] - block conv_transpose3d, giving error messages

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

diff --git a/aten/src/ATen/native/mps/operations/Convolution.mm b/aten/src/ATen/native/mps/operations/Convolution.mm
index 7336eabb27ee6..20432d2933e65 100644
--- a/aten/src/ATen/native/mps/operations/Convolution.mm
+++ b/aten/src/ATen/native/mps/operations/Convolution.mm
@@ -509,6 +509,7 @@ Tensor _mps_convolution_transpose(
     const Tensor& input_t, const Tensor& weight_t,
     IntArrayRef padding, IntArrayRef output_padding, 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);