re-enabling all distributed unit tests

NVIDIA · Apr 29, 2024 · 267116e · 267116e
1 parent e4c8cb3
commit 267116e
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/tests/test_distributed_convolution.py b/tests/test_distributed_convolution.py
@@ -176,18 +176,18 @@ def _gather_helper_bwd(self, tensor, B, C, convolution_dist):
 
     @parameterized.expand(
         [
-            # [128, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-6],
+            [128, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-6],
             [129, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-6],
-            # [128, 256, 128, 256, 32, 8, [3, 2], 1, "equiangular", "equiangular", False, 1e-6],
-            # [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],
-            # [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], 1, "equiangular", "equiangular", False, 1e-6],
+            [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],
+            [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],
         ]
     )
     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):