From 0d82bdc0e8eef1abea3d7adbc153dac2916db793 Mon Sep 17 00:00:00 2001 From: Predrag Ilkic Date: Wed, 13 Nov 2024 17:12:04 +0100 Subject: [PATCH] [test] removing xfail in test_ops.py I've noticed some tests are now passing (with latest tt-mlir/tt-metal), so removing `xfail` mark for those. --- forge/test/mlir/test_ops.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/forge/test/mlir/test_ops.py b/forge/test/mlir/test_ops.py index d91a3ea6..1ea3d50a 100644 --- a/forge/test/mlir/test_ops.py +++ b/forge/test/mlir/test_ops.py @@ -129,7 +129,6 @@ def get_input_tensor(dtype): (1, 7, 256), ], ) -@pytest.mark.xfail(reason="Found Unsupported operations while lowering from TTForge to TTIR in forward graph") @pytest.mark.push def test_sin(shape): class sin(nn.Module): @@ -158,7 +157,6 @@ def forward(self, x): (1, 7, 256), ], ) -@pytest.mark.xfail(reason="Found Unsupported operations while lowering from TTForge to TTIR in forward graph") @pytest.mark.push def test_cosine(shape): class cosine(nn.Module): @@ -1070,9 +1068,6 @@ def forward(self, x, y): assert compare_with_golden_pcc(golden=fw_out, calculated=co_out[0], pcc=0.99) -@pytest.mark.xfail( - reason="Unable to reshape a tensor in TILE_LAYOUT to non-tile height and width! Please convert the tensor to ROW_MAJOR_LAYOUT first" -) @pytest.mark.parametrize("x_shape", [7, 32, 41]) @pytest.mark.parametrize("y_shape", [7, 32, 41]) @pytest.mark.parametrize("dim", [1, 2]) @@ -1085,6 +1080,11 @@ def __init__(self): def forward(self, x): return torch.mean(x, dim=dim) + if dim == 1 and y_shape == 32: + pytest.xfail( + "TTNN: tilize fails for float32 - will be fixed once a080e2f035990d57ce5436a8affb3f052a5a1b5f in tt-metal is consumed" + ) + inputs = [ torch.rand(1, x_shape, y_shape), ]