diff --git a/python/tests/cuda_skip.py b/python/tests/cuda_skip.py index 25468082f..5bb465e1e 100644 --- a/python/tests/cuda_skip.py +++ b/python/tests/cuda_skip.py @@ -19,6 +19,7 @@ cuda_skip = { "TestConv.test_conv_general_flip_grad", "TestConv.test_conv_groups_grad", "TestConv.test_torch_conv_1D_grad", + "TestConv.test_torch_conv_2D", "TestConv.test_torch_conv_2D_grad", "TestConv.test_torch_conv_3D_grad", "TestConv.test_torch_conv_depthwise",