Skip to content

Commit d7ea1a0

Browse files
Merge branch 'main' into add-cast-int64-placeholders-to-int32-pass
2 parents fbfc2b5 + 29cec35 commit d7ea1a0

34 files changed

+701
-670
lines changed

backends/arm/_passes/arm_pass_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,7 @@ def _tosa_FP_pipeline(self, exported_program: ExportedProgram) -> GraphModule:
197197
self.add_pass(CastBoolToInt8Pass())
198198
self.add_pass(DecomposeSinhPass())
199199
self.add_pass(DecomposeSignPass())
200+
self.add_pass(DecomposeDivTensorModePass())
200201
self.add_pass(ReplaceScalarWithTensorArgPassTOSAMI())
201202
self.add_pass(DecomposeEmbeddingPass())
202203
self.add_pass(FuseQuantizedActivationPass())
@@ -215,7 +216,6 @@ def _tosa_FP_pipeline(self, exported_program: ExportedProgram) -> GraphModule:
215216
DecomposeMeanDimPass(exported_program.graph_module, self.tosa_spec)
216217
)
217218
self.add_pass(DecomposeNotEqualPass())
218-
self.add_pass(DecomposeDivTensorModePass())
219219
self.add_pass(DecomposeDivPass())
220220
self.add_pass(DecomposeSoftmaxPass())
221221
self.add_pass(DecomposeGeluPass())
@@ -285,6 +285,7 @@ def transform_for_annotation_pipeline(self, graph_module: GraphModule):
285285
self.add_pass(CastBoolToInt8Pass())
286286
self.add_pass(DecomposeSignPass())
287287
self.add_pass(DecomposeAddmmPass())
288+
self.add_pass(DecomposeDivTensorModePass())
288289
self.add_pass(ReplaceScalarWithTensorArgPassTOSABI())
289290
self.add_pass(ScalarsToAttributePass())
290291
self.add_pass(DecomposeGroupNormPass())
@@ -294,7 +295,6 @@ def transform_for_annotation_pipeline(self, graph_module: GraphModule):
294295
self.add_pass(DecomposeNotEqualPass())
295296
self.add_pass(DecomposeCosineSimilarityPass())
296297
self.add_pass(DecomposeGluPass())
297-
self.add_pass(DecomposeDivTensorModePass())
298298
self.add_pass(DecomposeDivPass())
299299
self.add_pass(DecomposeLeakyReLUPass())
300300
self.add_pass(DecomposeLinearVectorNormPass())

backends/arm/arm_backend.py

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

0 commit comments

Comments
 (0)