@@ -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 ())
0 commit comments