Uplift third_party/tt-mlir to origin/main 2025-01-14 (#1039)

This PR uplifts the third_party/tt-mlir submodule to the origin/main --------- Co-authored-by: Predrag Ilkic <[email protected]>
tenstorrent · Jan 14, 2025 · 3d75885 · 3d75885
1 parent 7866dfe
commit 3d75885
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 6 deletions.
diff --git a/forge/test/mlir/llama/tests/test_specific_ops_llama32.py b/forge/test/mlir/llama/tests/test_specific_ops_llama32.py
@@ -138,7 +138,7 @@ def forward(self, x):
         ((32, 11, 64), (32, 64, 11)),
         ((32, 11, 11), (32, 11, 64)),
         ((11, 2048), (2048, 8192)),
-        pytest.param(((1, 11, 8192), (8192, 2048)), marks=pytest.mark.xfail(reason="pcc ~ 0.65")),
+        ((1, 11, 8192), (8192, 2048)),
         ((1, 11, 2048), (2048, 128256)),
     ],
 )

diff --git a/forge/test/mlir/mnist/training/test_training.py b/forge/test/mlir/mnist/training/test_training.py
@@ -146,7 +146,7 @@ def test_mnist_training_with_grad_accumulation():
             total_loss += loss.item()
 
             golden_loss = loss_fn(golden_pred, target)
-            assert torch.allclose(loss, golden_loss, rtol=1e-1)  # 10% tolerance
+            assert torch.allclose(loss, golden_loss, rtol=0.13)
 
             # Run backward pass on device
             loss.backward()

diff --git a/forge/test/mlir/resnet/test_resnet_unique_ops.py b/forge/test/mlir/resnet/test_resnet_unique_ops.py
@@ -82,9 +82,6 @@ def forward(self, x):
             1,  # Outer dimension x
             1000,  # Outer dimension y
             2048,  # Inner dimension
-            marks=pytest.mark.xfail(
-                reason="Tensor mismatch. PCC = 0.9425581505871167, but required = 0.99. Tracking on: https://github.com/tenstorrent/tt-mlir/issues/1576"
-            ),
         ),
     ],
 )

diff --git a/third_party/tt-mlir b/third_party/tt-mlir
+1 −1		lib/OpModel/TTNN/Conversion.cpp
+1 −1		runtime/lib/ttnn/include/tt/runtime/ttnn/utils.cpp
+1 −1		runtime/lib/ttnn/operations/include/tt/runtime/ttnn/operations/utils.cpp
+0 −1		test/unittests/OpModel/TTNN/Conversion/TestConversion.cpp
+1 −1		third_party/CMakeLists.txt