From b8060aeadf2c9904b4a67284c6d41642f1434f4d Mon Sep 17 00:00:00 2001 From: Tom Allsop Date: Mon, 3 Feb 2025 14:28:17 +0000 Subject: [PATCH] Prioritise FuseBatchnorm2D over DecomposeBatchNormPass * As it is preferable to fuse batch_norm over decomposing it, prioritise this pass higher Signed-off-by: Tom Allsop Change-Id: I353d324ca31b85f1c62c866c93b49253ab7b4014 --- backends/arm/_passes/arm_pass_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/arm/_passes/arm_pass_manager.py b/backends/arm/_passes/arm_pass_manager.py index 686bfbcd8af..0c3d4f31aac 100644 --- a/backends/arm/_passes/arm_pass_manager.py +++ b/backends/arm/_passes/arm_pass_manager.py @@ -123,6 +123,7 @@ def _tosa_080_MI_pipeline(self, exported_program: ExportedProgram) -> GraphModul self.add_pass(FuseQuantizedActivationPass()) self.add_pass(RemoveGetItemPass()) self.add_pass(ConvertSplitToSlicePass()) + self.add_pass(FuseBatchnorm2DPass(exported_program)) self.add_pass(ConvertMmToBmmPass()) self.add_pass(DecomposeLinearPass()) self.add_pass(DecomposeBatchNormPass()) @@ -132,7 +133,6 @@ def _tosa_080_MI_pipeline(self, exported_program: ExportedProgram) -> GraphModul self.add_pass(ConvertMeanDimToAveragePoolPass()) self.add_pass(DecomposeDivPass()) self.add_pass(DecomposeSoftmaxesPass()) - self.add_pass(FuseBatchnorm2DPass(exported_program)) self.add_pass(AnnotateDecomposedMatmulPass()) self.add_pass(QuantizeOperatorArguments())