Skip to content

Commit

Permalink
Disable upsample_nearest 2d decomposition, nightly tests were broken …
Browse files Browse the repository at this point in the history
…due to it (#145)
  • Loading branch information
AleksKnezevic committed Dec 26, 2024
1 parent c9ba0ee commit bdeffd5
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
3 changes: 3 additions & 0 deletions tests/torch/test_interpolation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion tt_torch/dynamo/decompositions.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
}

Expand Down

0 comments on commit bdeffd5

Please sign in to comment.