Skip to content

Arm backend: Remove FuseConsecutiveRescalesPass #20146

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gggekov wants to merge 1 commit into
base: main
Choose a base branch
from
+2 −428
Open

Arm backend: Remove FuseConsecutiveRescalesPass #20146

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion backends/arm/_passes/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,6 @@
)
from .fuse_batch_norm2d_pass import FuseBatchNorm2dPass # noqa
from .fuse_consecutive_concat_shapes import FuseConsecutiveConcatShapesPass # noqa
from .fuse_consecutive_rescales_pass import FuseConsecutiveRescalesPass # noqa
from .fuse_constant_ops_pass import ( # noqa
ComputeConstantOpsAOTPass,
FuseConstantArgsPass,
Expand Down
2 changes: 0 additions & 2 deletions backends/arm/_passes/arm_pass_manager.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,6 @@
FoldAndAnnotateQParamsPass,
FuseBatchNorm2dPass,
FuseConsecutiveConcatShapesPass,
FuseConsecutiveRescalesPass,
FuseConstantArgsPass,
FuseDuplicateUsersPass,
FuseEqualPlaceholdersPass,
Expand Down Expand Up @@ -417,7 +416,6 @@ def _tosa_pipeline(
# Ticket: MLETORCH-1539
DecomposeLinearPass(),
InsertRescaleInt32Pass(),
FuseConsecutiveRescalesPass(),
InsertControlFlowRescalesPass(),
DecomposeQuantNodesPass(),
]
Expand Down
175 changes: 0 additions & 175 deletions backends/arm/_passes/fuse_consecutive_rescales_pass.py
View file
Open in desktop

This file was deleted.

8 changes: 2 additions & 6 deletions backends/arm/test/models/test_residual_conv_block.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@
"""Residual conv block model test for ARM TOSA backend.
Tests a minimal residual architecture with conv->batchnorm->relu->add blocks and
permute operations, representative of quantized signal processing models where
FuseConsecutiveRescalesPass eliminates redundant RESCALE pairs.
permute operations.
"""

Expand All @@ -28,10 +27,7 @@ class ResidualConvBlock(torch.nn.Module):
Architecture: conv->bn->relu->add (residual) -> permute ->
conv->bn->relu->add. When quantized, each residual add is
wrapped with INT32 RESCALEs by InsertRescaleInt32Pass. Stacked
blocks create consecutive RESCALE pairs (INT32->INT8->INT32)
between adjacent adds that FuseConsecutiveRescalesPass
eliminates.
wrapped with INT32 RESCALEs by InsertRescaleInt32Pass.
"""

Expand Down
Loading
Loading