Skip to content

Commit b9a0d81

Browse files
Martin LindströmMartin Lindström
authored andcommitted
Arm backend: Remove DecomposeLinearVectorNormPass from INT pipeline
DecomposeLinearVectorNormPass is listed in both the transform_for_annotation_pipeline and _tosa_INT_pipeline stages. The latter is redundant because torch.linalg.vector_norm can only run with floating point inputs, i.e., we should always be in a quantized setting when we enter _tosa_INT_pipeline if the operator is present and _transform_for_annotation_pipeline will always run; therefore, remove DecomposeLinearVectorNormPass from _tosa_INT_pipeline. Signed-off-by: Martin Lindström <[email protected]> Change-Id: I647687b51298bbb98087914fbbee053436ffb79f
1 parent 3371b5e commit b9a0d81

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

backends/arm/_passes/arm_pass_manager.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,6 @@ def _tosa_INT_pipeline(self, exported_program: ExportedProgram) -> GraphModule:
155155
self.add_pass(RemoveGetItemPass())
156156
self.add_pass(ConvertSplitToSlicePass())
157157
self.add_pass(ConvertMmToBmmPass())
158-
self.add_pass(DecomposeLinearVectorNormPass())
159158
self.add_pass(
160159
DecomposeMeanDimPass(exported_program.graph_module, self.tosa_spec)
161160
)

0 commit comments

Comments
 (0)