Skip to content

Commit b1e6617

Browse files
authored
Fix pyre
Differential Revision: D65684570 Pull Request resolved: #6740
1 parent ddc8ea6 commit b1e6617

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

backends/arm/_passes/TARGETS

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ python_library(
77
deps = [
88
"//executorch/backends/arm:tosa_quant_utils",
99
"//executorch/backends/arm:tosa_utils",
10+
"//executorch/backends/xnnpack/_passes:xnnpack_passes",
1011
"//executorch/exir:lib",
1112
],
1213
)

backends/arm/operators/op_max_pool2d.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
# LICENSE file in the root directory of this source tree.
55

66
# pyre-unsafe
7-
from typing import cast, List
7+
from typing import List
88

99
import serializer.tosa_serializer as ts
1010
import torch
@@ -54,9 +54,7 @@ def define_node(
5454
output_zp = 0
5555

5656
if is_quant_node:
57-
input_zp = get_quant_node_args(
58-
cast(torch.fx.Node, node.all_input_nodes[0])
59-
).zp
57+
input_zp = get_quant_node_args(node.all_input_nodes[0]).zp
6058
output_zp = get_quant_node_args(list(node.users)[0]).zp
6159

6260
attr = ts.TosaSerializerAttribute()

0 commit comments

Comments
 (0)