diff --git a/tt_torch/dynamo/backend.py b/tt_torch/dynamo/backend.py index 814c3021..8ad9f749 100644 --- 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