Skip to content

Commit 7a4b5ab

Browse files
committed
Fix conv3d linter errors
Signed-off-by: Ryan O'Shea <[email protected]> Change-Id: I9b096c71da7e1bf943857c04c5606e22ba35da4f
1 parent 88ed89a commit 7a4b5ab

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

backends/arm/_passes/rewrite_conv_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def insert_output_rescale(self, graph_module, node):
218218
)
219219
return rescale_node
220220

221-
def call(self, graph_module: torch.fx.GraphModule) -> PassResult: # noqa: C901
221+
def call(self, graph_module: torch.fx.GraphModule) -> PassResult: # noqa: C901
222222
modified = False
223223
for node in graph_module.graph.nodes:
224224
if (

backends/arm/test/misc/test_dw_convs_with_shared_weights.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from typing import Any, Tuple
77

88
import torch
9-
from executorch.backends.arm._passes.rewrite_conv2d_pass import RewriteConv2dPass
9+
from executorch.backends.arm._passes.rewrite_conv_pass import RewriteConvPass
1010
from executorch.backends.arm.test.tester.test_pipeline import (
1111
PassPipeline,
1212
TosaPipelineFP,
@@ -51,7 +51,7 @@ def test_convs_tosa_int():
5151
def test_rewrite_conv_pass():
5252
module = DWConvsModule()
5353
pipeline = PassPipeline(
54-
module, module.get_inputs(), passes_with_exported_program=[RewriteConv2dPass]
54+
module, module.get_inputs(), passes_with_exported_program=[RewriteConvPass]
5555
)
5656
# We can't run TOSA backend dialect operators in eager mode
5757
pipeline.pop_stage("run_method_and_compare_outputs")

0 commit comments

Comments
 (0)