From f8fa59ef25ada8772ce6621d723ca5c8678814e9 Mon Sep 17 00:00:00 2001 From: Muhammad Asif Manzoor Date: Thu, 20 Feb 2025 21:07:38 +0000 Subject: [PATCH] Enable reduction tests * Reduction sum along 3 dims * Reduction mean along 3 dims --- tests/torch/test_reduction.py | 21 ++------------------- 1 file changed, 2 insertions(+), 19 deletions(-) diff --git a/tests/torch/test_reduction.py b/tests/torch/test_reduction.py index 5c5bef05..f7bd5301 100644 --- a/tests/torch/test_reduction.py +++ b/tests/torch/test_reduction.py @@ -22,16 +22,7 @@ ([(4, 2, 32, 32)], [3], False, [torch.bfloat16], 0.02), ([(4, 2, 32, 32)], [0, 2], False, [torch.bfloat16], 0.035), ([(4, 2, 32, 32)], [0, 1, 2, 3], True, [torch.bfloat16], 0.13), - pytest.param( - [(4, 2, 32, 32)], - [0, 2, 3], - True, - [torch.bfloat16], - 0.35, - marks=pytest.mark.xfail( - reason="Reduce on more than two dimensions is not currently supported by TTNN" - ), - ), + ([(4, 2, 32, 32)], [0, 2, 3], True, [torch.bfloat16], 0.30), ], ) def test_reduce_sum(input_shape, dim_arg, keep_dim, input_type, atol): @@ -202,15 +193,7 @@ def forward(self, x): ([(4, 2, 32, 32)], [0, 2], False, [torch.bfloat16]), ([(4, 2, 32, 32)], [1, 2], True, [torch.bfloat16]), ([(4, 2, 32, 32)], [0, 1, 2, 3], True, [torch.bfloat16]), - pytest.param( - [(4, 2, 32, 32)], - [1, 2, 3], - True, - [torch.bfloat16], - marks=pytest.mark.xfail( - reason="Reduce on more than two dimensions is not currently supported by TTNN" - ), - ), + ([(4, 2, 32, 32)], [1, 2, 3], True, [torch.bfloat16]), ], ) def test_reduce_mean(input_shape, dim_arg, keep_dim, input_type):