From 32f0ae02c85e6c678296b8c78fb5c7bbe4a466d3 Mon Sep 17 00:00:00 2001 From: Boris Bonev Date: Mon, 29 Apr 2024 12:05:01 +0200 Subject: [PATCH] relaxing tolerances in distributed test --- tests/test_distributed_convolution.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/tests/test_distributed_convolution.py b/tests/test_distributed_convolution.py index 3b66048..90e4a22 100644 --- 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], - # [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], 1, "equiangular", "equiangular", False, 1e-5], + [129, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-5], + [128, 256, 128, 256, 32, 8, [3, 2], 1, "equiangular", "equiangular", False, 1e-5], + [128, 256, 64, 128, 32, 8, [3], 1, "equiangular", "equiangular", False, 1e-5], + [128, 256, 128, 256, 32, 8, [3], 2, "equiangular", "equiangular", False, 1e-5], + [128, 256, 128, 256, 32, 5, [3], 1, "equiangular", "equiangular", False, 1e-5], + [128, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-5], + [129, 256, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-5], + [128, 256, 128, 256, 32, 8, [3, 2], 1, "equiangular", "equiangular", True, 1e-5], + [64, 128, 128, 256, 32, 8, [3], 1, "equiangular", "equiangular", True, 1e-5], + [128, 256, 128, 256, 32, 8, [3], 2, "equiangular", "equiangular", True, 1e-5], + [128, 256, 128, 256, 32, 5, [3], 1, "equiangular", "equiangular", True, 1e-5], ] ) 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):