narrowing down tests

NVIDIA · Apr 29, 2024 · 40a10f5 · 40a10f5
1 parent db37c5f
commit 40a10f5
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_distributed_convolution.py b/tests/test_distributed_convolution.py
@@ -182,12 +182,12 @@ def _gather_helper_bwd(self, tensor, B, C, convolution_dist):
             # [128, 256, 64, 128, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-6],
             # [128, 256, 128, 256, 32, 8, [3], 2, "equiangular", "equiangular", False, 1e-6],
             # [128, 256, 128, 256, 32, 5, [3], 1, "equiangular", "equiangular", False, 1e-6],
-            [128, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-6],
-            [129, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-6],
-            [128, 256, 128, 256, 32, 8, [3, 2], 1, "equiangular", "equiangular", True, 1e-6],
+            # [128, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-6],
+            # [129, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-6],
+            # [128, 256, 128, 256, 32, 8, [3, 2], 1, "equiangular", "equiangular", True, 1e-6],
             [64, 128, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-6],
-            [128, 256, 128, 256, 32, 8, [3], 2, "equiangular", "equiangular", True, 1e-6],
-            [128, 256, 128, 256, 32, 5, [3], 1, "equiangular", "equiangular", True, 1e-6],
+            # [128, 256, 128, 256, 32, 8, [3], 2, "equiangular", "equiangular", True, 1e-6],
+            # [128, 256, 128, 256, 32, 5, [3], 1, "equiangular", "equiangular", True, 1e-6],
         ]
     )
     def test_distributed_disco_conv(self, nlat_in, nlon_in, nlat_out, nlon_out, batch_size, num_chan, kernel_shape, groups, grid_in, grid_out, transpose, tol):