5151 RetraceFoldedDtypesPass ,
5252)
5353from executorch .backends .arm ._passes .fuse_batchnorm2d_pass import FuseBatchnorm2DPass
54- from executorch .backends .arm ._passes .fuse_constant_ops_pass import FuseConstantOpsPass
5554from executorch .backends .arm ._passes .fuse_quantized_activation_pass import ( # type: ignore[import-not-found]
5655 FuseQuantizedActivationPass ,
5756)
7978 UnsqueezeScalarPlaceholdersPass ,
8079)
8180from executorch .backends .arm .tosa_specification import TosaSpecification
82- from executorch .backends .transforms .fuse_view_copy import FuseViewCopyTransform
8381
8482from 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
0 commit comments