Skip to content

Commit

Permalink
Uplift third_party/tt-mlir to 2025-01-10 (#183)
Browse files Browse the repository at this point in the history
This PR uplifts the third_party/tt-mlir to the

---------

Co-authored-by: uazizTT <[email protected]>
Co-authored-by: Usman Aziz <[email protected]>
Co-authored-by: brataTT <[email protected]>
  • Loading branch information
4 people authored Jan 10, 2025
1 parent 1b4b6b9 commit b73eff6
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 6 deletions.
5 changes: 3 additions & 2 deletions tests/models/autoencoder_linear/test_autoencoder_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,11 +94,12 @@ def test_autoencoder_linear(record_property, mode, nightly):
cc = CompilerConfig()
cc.enable_consteval = True
cc.consteval_parameters = True
cc.remove_embedded_constants = True
if nightly:
cc.compile_depth = CompileDepth.EXECUTE_OP_BY_OP

tester = ThisTester(model_name, mode, compiler_config=cc)
tester = ThisTester(
model_name, mode, pcc=0.1, required_atol=0.2, compiler_config=cc
)
results = tester.test_model()

if mode == "eval":
Expand Down
8 changes: 5 additions & 3 deletions tests/torch/test_basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,10 @@ def forward(self, x):
verify_module(Basic(), input_shapes=[(32, 32)])


@pytest.mark.xfail(
strict=True,
reason="RepeatOp is failing due to scalar type shape mismatch, see https://github.com/tenstorrent/tt-torch/issues/186",
)
def test_linear_with_bias_no_embedded_constants():
class Basic(nn.Module):
def __init__(self):
Expand All @@ -225,9 +229,7 @@ def __init__(self):
def forward(self, x):
return x + 1.0

cc = CompilerConfig()
cc.remove_embedded_constants = True
verify_module(Basic(), input_shapes=[(1, 768)], compiler_config=cc)
verify_module(Basic(), input_shapes=[(1, 768)])


def test_maximum():
Expand Down
2 changes: 1 addition & 1 deletion third_party/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#

set(TT_MLIR_VERSION "f6b5753630b6f2005f559d3ce223a72d94bf01d5")
set(TT_MLIR_VERSION "2fcd37aa351ea7bb1baae5869546e2d0752505ce")

if (TOOLCHAIN STREQUAL "ON")
cmake_minimum_required(VERSION 3.20)
Expand Down

0 comments on commit b73eff6

Please sign in to comment.