diff --git a/backends/arm/_passes/TARGETS b/backends/arm/_passes/TARGETS index ca20b03fccd..6ca59cfee27 100644 --- a/backends/arm/_passes/TARGETS +++ b/backends/arm/_passes/TARGETS @@ -7,6 +7,7 @@ python_library( deps = [ "//executorch/backends/arm:tosa_quant_utils", "//executorch/backends/arm:tosa_utils", + "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/exir:lib", ], ) diff --git a/backends/arm/operators/op_max_pool2d.py b/backends/arm/operators/op_max_pool2d.py index 0752d8242f7..a0b868f684d 100644 --- a/backends/arm/operators/op_max_pool2d.py +++ b/backends/arm/operators/op_max_pool2d.py @@ -4,7 +4,7 @@ # LICENSE file in the root directory of this source tree. # pyre-unsafe -from typing import cast, List +from typing import List import serializer.tosa_serializer as ts import torch @@ -54,9 +54,7 @@ def define_node( output_zp = 0 if is_quant_node: - input_zp = get_quant_node_args( - cast(torch.fx.Node, node.all_input_nodes[0]) - ).zp + input_zp = get_quant_node_args(node.all_input_nodes[0]).zp output_zp = get_quant_node_args(list(node.users)[0]).zp attr = ts.TosaSerializerAttribute()