From bdeffd57e21b819d910eb69c32b4e85b5d4f49e9 Mon Sep 17 00:00:00 2001 From: Aleks Knezevic Date: Thu, 26 Dec 2024 01:44:21 +0000 Subject: [PATCH] Disable upsample_nearest 2d decomposition, nightly tests were broken due to it (#145) --- tests/torch/test_interpolation.py | 3 +++ tt_torch/dynamo/decompositions.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/torch/test_interpolation.py b/tests/torch/test_interpolation.py index 531a2f83..0c953102 100644 --- a/tests/torch/test_interpolation.py +++ b/tests/torch/test_interpolation.py @@ -46,6 +46,9 @@ def forward(self, x): @pytest.mark.parametrize("inW", [50, 128, 224, 540]) @pytest.mark.parametrize("inC", [1, 3]) @pytest.mark.parametrize("scale_factor", [2, 3]) +@pytest.mark.xfail( + reason=" #TODO: https://github.com/tenstorrent/tt-torch/issues/145" +) def test_nearest_upsample(inH, inW, inC, scale_factor): class Interpolate(nn.Module): def __init__(self): diff --git a/tt_torch/dynamo/decompositions.py b/tt_torch/dynamo/decompositions.py index f4b0ca2c..bafd450d 100644 --- a/tt_torch/dynamo/decompositions.py +++ b/tt_torch/dynamo/decompositions.py @@ -235,7 +235,7 @@ def _get_default_decomposition_ops() -> DecompositionOpsList: def _get_custom_decopositions() -> DecompositionTable: aten = torch.ops.aten return { - aten.upsample_nearest2d.default: upsample_nearest2d, + # aten.upsample_nearest2d.default: upsample_nearest2d, #TODO: https://github.com/tenstorrent/tt-torch/issues/145 aten.upsample_bilinear2d.default: upsample_bilinear2d, }