Merge branch 'main' into kw/gen_acc_tests

tenstorrent · Jan 23, 2025 · acdc5e8 · acdc5e8
2 parents 278e8e2 + 9d89f70
commit acdc5e8
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 3 deletions.
diff --git a/.github/workflows/update-ttnn-wheel.yaml b/.github/workflows/update-ttnn-wheel.yaml
@@ -30,7 +30,7 @@ jobs:
           latest_version=${{ steps.fetch_release.outputs.release }}
           latest_version_short=$(echo $latest_version | sed 's/-rc/rc/')          
           sed -i '/^https:\/\/github\.com\/tenstorrent\/tt-metal\/releases\//d' requirements.txt          
-          echo "https://github.com/tenstorrent/tt-metal/releases/download/v$latest_version/ttnn-$latest_version_short+wormhole.b0-cp38-cp38-linux_x86_64.whl" >> requirements.txt
+          echo "https://github.com/tenstorrent/tt-metal/releases/download/v$latest_version/ttnn-$latest_version_short+any-cp38-cp38-linux_x86_64.whl" >> requirements.txt
 
       - name: Create Pull Request
         uses: peter-evans/create-pull-request@v7

diff --git a/requirements.txt b/requirements.txt
@@ -6,4 +6,4 @@ tabulate==0.9.0
 networkx==3.1
 graphviz
 matplotlib==3.7.1
-https://github.com/tenstorrent/tt-metal/releases/download/v0.54.0-rc23/ttnn-0.54.0rc23+wormhole.b0-cp38-cp38-linux_x86_64.whl
+https://github.com/tenstorrent/tt-metal/releases/download/v0.55.0-rc12/ttnn-0.55.0rc12+any-cp38-cp38-linux_x86_64.whl
diff --git a/tests/lowering/creation/test_arange.py b/tests/lowering/creation/test_arange.py
@@ -90,6 +90,6 @@ def test_arange_start_step(device, input_shapes):
 
     # Check the graph has be rewritten and contain ttnn ops
     nodes = list(option._out_fx_graphs[0].nodes)
-    assert [node.target for node in nodes].count(ttnn.arange) == 1
+    assert [node.target for node in nodes].count(ttnn.arange) == 1 or [node.op for node in nodes].count("get_attr")
     # Check inference result
     assert torch.allclose(result_before, result_after)
diff --git a/torch_ttnn/passes/constant_folding_pass.py b/torch_ttnn/passes/constant_folding_pass.py
@@ -11,6 +11,7 @@ def __init__(self):
             torch.ops.aten.lift_fresh_copy.default,
             torch.ops.aten.pow.Tensor_Tensor,
             torch.ops.aten.arange.start,
+            torch.ops.aten.arange.start_step,
             torch.ops.aten.unsqueeze.default,
             torch.ops.aten.arange.default,
             torch.ops.aten.view.default,

diff --git a/torch_ttnn/passes/lowering/to_tt_guard.py b/torch_ttnn/passes/lowering/to_tt_guard.py
@@ -83,6 +83,13 @@
     ["List[Tensor] tensors = [<[13600]>, <[13600]>, <[13600]>, <[13600]>]", "int dim = 1"],
 ]
 
+############################################################
+# EXTRA BLOCKLIST OF falcon-7b-instruct
+############################################################
+# RuntimeError: TT_THROW @ /tmp/build-via-sdist-d26xvola/ttnn-0.54.0rc18+wormhole.b0/ttnn/cpp/ttnn/device_operation.hpp:487: tt::exception
+# Unsupported storage type
+aten_argmax_default_blocklist += [["Tensor<[1, 7]> self = ?", "Optional[int] dim = 1", "bool keepdim = True"]]
+
 ############################################################
 # EXTRA BLOCKLIST OF retinanet_resnet50_fpn_v2
 ############################################################