Skip to content

Commit 31e4bfc

Browse files
authored
Revert "Arm backend: Add FuseViewCopyTransform and FuseConstantsPass in arm_p…"
This reverts commit 3a7c231.
1 parent 56c94c2 commit 31e4bfc

File tree

4 files changed

+2
-319
lines changed

4 files changed

+2
-319
lines changed

backends/arm/_passes/arm_pass_manager.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@
5151
RetraceFoldedDtypesPass,
5252
)
5353
from executorch.backends.arm._passes.fuse_batchnorm2d_pass import FuseBatchnorm2DPass
54-
from executorch.backends.arm._passes.fuse_constant_ops_pass import FuseConstantOpsPass
5554
from executorch.backends.arm._passes.fuse_quantized_activation_pass import ( # type: ignore[import-not-found]
5655
FuseQuantizedActivationPass,
5756
)
@@ -79,7 +78,6 @@
7978
UnsqueezeScalarPlaceholdersPass,
8079
)
8180
from executorch.backends.arm.tosa_specification import TosaSpecification
82-
from executorch.backends.transforms.fuse_view_copy import FuseViewCopyTransform
8381

8482
from executorch.backends.transforms.replace_scalar_with_tensor import (
8583
ReplaceScalarWithTensorArgPass,
@@ -116,6 +114,7 @@ def _tosa_080_BI_pipeline(self, exported_program: ExportedProgram) -> GraphModul
116114
self.add_pass(QuantizeOperatorArguments())
117115
self.add_pass(FoldAndAnnotateQParamsPass()) # type: ignore[call-arg]
118116
self.add_pass(RetraceFoldedDtypesPass())
117+
self.add_pass(InsertTableOpsPass(exported_program))
119118

120119
self.add_pass(RemoveClonePass())
121120
self.add_pass(SizeAdjustConv2DPass())
@@ -129,12 +128,8 @@ def _tosa_080_BI_pipeline(self, exported_program: ExportedProgram) -> GraphModul
129128
self.add_pass(DecomposeSelectPass())
130129
self.add_pass(ConvertSqueezesToViewPass())
131130

132-
self.add_pass(FuseViewCopyTransform())
133-
self.add_pass(FuseConstantOpsPass(exported_program))
134-
self.add_pass(InsertTableOpsPass(exported_program))
135131
self.add_pass(AnnotateChannelsLastDimOrder())
136132
self.add_pass(InsertRescalePass())
137-
138133
return self._transform(exported_program.graph_module)
139134

140135
def _tosa_080_MI_pipeline(self, exported_program: ExportedProgram) -> GraphModule:
@@ -160,6 +155,7 @@ def _tosa_080_MI_pipeline(self, exported_program: ExportedProgram) -> GraphModul
160155
self.add_pass(QuantizeOperatorArguments())
161156
self.add_pass(FoldAndAnnotateQParamsPass()) # type: ignore[call-arg]
162157
self.add_pass(RetraceFoldedDtypesPass())
158+
self.add_pass(InsertTableOpsPass(exported_program))
163159

164160
self.add_pass(RemoveClonePass())
165161
self.add_pass(SizeAdjustConv2DPass())
@@ -173,9 +169,6 @@ def _tosa_080_MI_pipeline(self, exported_program: ExportedProgram) -> GraphModul
173169
self.add_pass(DecomposeSelectPass())
174170
self.add_pass(ConvertSqueezesToViewPass())
175171

176-
self.add_pass(FuseViewCopyTransform())
177-
self.add_pass(FuseConstantOpsPass(exported_program))
178-
self.add_pass(InsertTableOpsPass(exported_program))
179172
self.add_pass(AnnotateChannelsLastDimOrder())
180173
self.add_pass(InsertRescalePass())
181174

backends/arm/_passes/arm_pass_utils.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
)
2727
from torch._ops import OpOverload
2828
from torch._subclasses.fake_tensor import FakeTensor
29-
from torch.export.graph_signature import InputKind
3029

3130

3231
def is_get_attr_node(node: torch.fx.Node) -> bool:
@@ -45,30 +44,6 @@ def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool:
4544
)
4645

4746

48-
def get_constant_placeholder_kind(
49-
exp_prog: ExportedProgram, node: torch.fx.Node
50-
) -> InputKind:
51-
if is_param(exp_prog, node):
52-
return InputKind.PARAMETER
53-
if is_buffer(exp_prog, node):
54-
return InputKind.BUFFER
55-
if is_lifted_tensor_constant(exp_prog, node):
56-
return InputKind.CONSTANT_TENSOR
57-
58-
raise RuntimeError("Node is neither PARAMETER, BUFFER nor CONSTANT_TENSOR")
59-
60-
61-
def is_persistent_buffer(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool | None:
62-
if is_buffer(exp_prog, node):
63-
buffer_name = exp_prog.graph_signature.inputs_to_buffers[node.name]
64-
if buffer_name in exp_prog.graph_signature.non_persistent_buffers:
65-
return False
66-
else:
67-
return True
68-
69-
return None
70-
71-
7247
def get_param_tensor(
7348
exp_prog: ExportedProgram, node: torch.fx.Node
7449
) -> Optional[torch.Tensor]:

backends/arm/_passes/fuse_constant_ops_pass.py

Lines changed: 0 additions & 170 deletions
This file was deleted.

backends/arm/test/passes/test_fuse_constant_ops_pass.py

Lines changed: 0 additions & 115 deletions
This file was deleted.

0 commit comments

Comments
 (0)