From 7dd3880cc44d325f2fb69a8fb093c0a028ea9914 Mon Sep 17 00:00:00 2001 From: Hansong Zhang Date: Fri, 8 Nov 2024 13:40:00 -0800 Subject: [PATCH] Fix pyre Differential Revision: D65684570 --- backends/arm/_passes/TARGETS | 1 + backends/arm/operators/op_max_pool2d.py | 6 ++---- 2 files changed, 3 insertions(+), 4 deletions(-) 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()