Skip to content

Commit dafd5f4

Browse files
eigen-kfacebook-github-bot
authored andcommitted
Use cadence remove ops passes in turing partitioner. (#12791)
Summary: Jarvis&Helios passes unification. Context: https://docs.google.com/document/d/1v3B-0ngdeEx0VoI8pAKovqeLkkByMM6hFBqtOzBboOw/edit?usp=sharing Differential Revision: D78854033
1 parent 37e3003 commit dafd5f4

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

backends/cadence/aot/remove_ops.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
import logging
2121
from dataclasses import dataclass, field
22-
from typing import cast, List, Optional, Sequence, Set
22+
from typing import cast, List, Optional, Sequence, Set, Type
2323

2424
import torch
2525
import torch.fx
@@ -940,21 +940,28 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
940940
# in Jarvis. Currently, each function in this class iterates over each node of
941941
# the graph module once. In future, we could consolidate them into a monolithic
942942
# function.
943-
class CadenceRemoveNops:
944-
passes = [
943+
class GenericRemoveNops:
944+
passes: List[Type[ExportPass]] = [
945945
SimplifySliceOpPass,
946946
RemoveCloneOpsTransformImported,
947947
RemoveToOpsPass,
948948
RemoveNopRequantizeOpPass,
949949
RemoveZeroSizedCatArgsPass,
950950
RemoveNopSliceOrViewOpPass,
951951
RemoveNopExpandOpPass,
952-
RemoveZeroSizedConstantPadNd,
953952
RemoveCloneOpPass,
954953
RemoveContiguousOpPass,
955954
RemoveAliasCopyOpPass,
956955
RemoveNopMulOpPass,
957956
RemoveNopAddOpPass,
958957
RemoveNopLinalgVectorNormOpPass,
959958
RemoveBranchedQuantDequant,
959+
RemoveNopSelectOpPass,
960+
RemovePermutesAroundElementwiseOps,
961+
RemoveSqueezeViewBeforeElementwiseOps,
962+
RemoveCatFromSliceCopyPass,
960963
]
964+
965+
class CadenceRemoveNops:
966+
passes: List[Type[ExportPass]] = GenericRemoveNops.passes
967+
passes += [RemoveZeroSizedConstantPadNd]

0 commit comments

Comments
 (0)