Skip to content

Commit

Permalink
Uplift third_party/tt-mlir to origin/main 2024-12-27 (#978)
Browse files Browse the repository at this point in the history
This PR uplifts the third_party/tt-mlir submodule to the origin/main

---------

Co-authored-by: Stefan Djordjevic <[email protected]>
  • Loading branch information
vmilosevic and sdjordjevicTT authored Dec 27, 2024
1 parent 4355d7a commit fcb9b4c
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
12 changes: 6 additions & 6 deletions forge/test/mlir/test_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -1234,7 +1234,7 @@ def forward(self, a):
((64,), -1, True),
((4, 64), 0, True),
((32, 32), -2, True),
pytest.param((2, 32, 32), 0, True, marks=pytest.mark.xfail(reason="tt:exception Unsupported dim")),
((2, 32, 32), 0, True),
((1, 64, 32), 2, True),
((4, 32, 64), -2, True),
((4, 128, 128, 128), 0, True),
Expand Down Expand Up @@ -1263,7 +1263,7 @@ def forward(self, a):
-2,
False,
),
pytest.param((2, 32, 32), 0, False, marks=pytest.mark.xfail(reason="tt:exception Unsupported dim")),
((2, 32, 32), 0, False),
((1, 64, 32), 2, False),
((4, 32, 64), -2, False),
((4, 128, 128, 128), 0, False),
Expand Down Expand Up @@ -1306,7 +1306,7 @@ def forward(self, a):
((64,), -1, True),
((4, 64), 0, True),
((32, 32), -2, True),
pytest.param((2, 32, 32), 0, True, marks=pytest.mark.xfail(reason="tt:exception Unsupported dim")),
((2, 32, 32), 0, True),
((1, 64, 32), 2, True),
((4, 32, 64), -2, True),
((4, 128, 128, 128), 0, True),
Expand Down Expand Up @@ -1335,7 +1335,7 @@ def forward(self, a):
-2,
False,
),
pytest.param((2, 32, 32), 0, False, marks=pytest.mark.xfail(reason="tt:exception Unsupported dim")),
((2, 32, 32), 0, False),
((1, 64, 32), 2, False),
((4, 32, 64), -2, False),
((4, 128, 128, 128), 0, False),
Expand Down Expand Up @@ -1609,7 +1609,7 @@ def forward(self, input_tensor, indices):
((64,), -1, True),
((4, 64), 0, True),
((32, 32), -2, True),
pytest.param((2, 32, 32), 0, True, marks=pytest.mark.xfail(reason="tt:exception Unsupported dim")),
((2, 32, 32), 0, True),
((1, 64, 32), 2, True),
((4, 32, 64), -2, True),
((4, 128, 128, 128), 0, True),
Expand All @@ -1630,7 +1630,7 @@ def forward(self, input_tensor, indices):
),
((4, 64), 0, False),
((32, 32), -2, False),
pytest.param((2, 32, 32), 0, False, marks=pytest.mark.xfail(reason="keepdim=False is not supported")),
((2, 32, 32), 0, False),
((1, 64, 32), 2, False),
((4, 32, 64), -2, False),
((4, 128, 128, 128), 0, False),
Expand Down
2 changes: 1 addition & 1 deletion third_party/tt-mlir

0 comments on commit fcb9b4c

Please sign in to comment.