wip

tenstorrent · Nov 7, 2024 · 26ae272 · 26ae272
1 parent b2f2dc9
commit 26ae272
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/tt_torch/dynamo/backend.py b/tt_torch/dynamo/backend.py
@@ -321,14 +321,6 @@ def run_gm_op_by_op(self, *inputs):
                     if out_degree[arg] == 0 and arg.op != "output":
                         del node_to_tensor[arg]
                         out_degree.pop(arg)
-        # Handle any intermediaries left - might be redundant
-        intermediates = [
-            node
-            for node, users in out_degree.items()
-            if users == 0 and node.op != "output"
-        ]
-        for node in intermediates:
-            del node_to_tensor[node]
 
         self.compiler_config.save_unique_ops()
         return outputs