def remove_ops(
    module: torch.fx.GraphModule,
) -> torch.fx.GraphModule:
    """
    1. Remove clone, _unsafe_view node. #TODO Remove this func after functionalization is workable
    2. Remove inefficient op getitem(index=slice) P561572458
    """
    modified = False
    for n in module.graph.nodes:
        if n.op == "call_function" and n.target in (torch.ops.aten.clone.default,):
            modified = True
            node = n
            input_n = node.all_input_nodes[0]
            node.replace_all_uses_with(input_n)
    module.graph.eliminate_dead_code()
    module.recompile()
    for n in module.graph.nodes:
        if n.op == "call_function" and n.target in (
            torch.ops.aten._unsafe_view.default,
        ):
            modified = True
            node = n
            with module.graph.inserting_after(node):
                new_node = module.graph.create_node(
                    "call_function",
                    torch.ops.aten.reshape,
                    args=node.args,
                    kwargs=node.kwargs,
                )
                node.replace_all_uses_with(new_node)
                module.graph.erase_node(node)
    module.graph.eliminate_dead_code()
    module.recompile()
    return PassResult(module, modified)
1. Remove clone, _unsafe_view node. #TODO Remove this func after functionalization is workable
2. Remove inefficient op getitem(index=slice) P561572458
fx2ait/fx2ait/passes/lower_basic_pass_aten.py