diff --git a/tests/test_distributed_convolution.py b/tests/test_distributed_convolution.py index aae79ae..57f939c 100644 --- a/tests/test_distributed_convolution.py +++ b/tests/test_distributed_convolution.py @@ -176,12 +176,12 @@ 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], - [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", 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],