Skip to content

Commit

Permalink
Remove unrelated changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Groverkss committed Jan 20, 2024
1 parent 3af36b8 commit f595a01
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 13 deletions.
12 changes: 0 additions & 12 deletions python/shark_turbine/aot/passes/remove_alias.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,18 +10,6 @@
import operator as py_operator


class Test(torch.nn.Module):
def __init__(self):
super().__init__()

def forward(self, x):
unrolled = torch.unbind(x, 1)
return unrolled[0], unrolled[1], unrolled[2], unrolled[3]


trace: GraphModule = torch.fx.symbolic_trace(Test())


def remove_unbind(gm: GraphModule) -> GraphModule:
# Find all unbind nodes
unbind_nodes = []
Expand Down
1 change: 0 additions & 1 deletion python/shark_turbine/dynamo/passes.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@
torch.ops.aten._log_softmax_backward_data,
torch.ops.aten.lift_fresh_copy.default,
torch.ops.aten._unsafe_index.Tensor,
torch.ops.aten.sinc,
# decompositions added manually in this file
torch.ops.aten._scaled_dot_product_flash_attention.default,
]
Expand Down

0 comments on commit f595a01

Please sign in to comment.