Skip to content

Commit 75a9bdd

Browse files
lucylqfacebook-github-bot
authored andcommitted
Fix arm dependency (#6918)
Summary: Adding dependencies for #6851 Reviewed By: larryliu0820, kirklandsign Differential Revision: D66039382
1 parent d5f91bf commit 75a9bdd

File tree

3 files changed

+17
-1
lines changed

3 files changed

+17
-1
lines changed

backends/arm/TARGETS

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ python_library(
99
typing = True,
1010
deps = [
1111
":arm_backend",
12+
"//executorch/backends/arm/operator_support:operator_support",
1213
"//executorch/backends/arm/_passes:passes",
1314
"//executorch/exir:lib",
1415
],

backends/arm/operator_support/TARGETS

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
2+
3+
python_library(
4+
name = "operator_support",
5+
srcs = glob(["*.py"]),
6+
typing = True,
7+
deps = [
8+
"//executorch/backends/xnnpack/_passes:xnnpack_passes",
9+
"//executorch/exir:lib",
10+
"//executorch/backends/arm:tosa_specification"
11+
],
12+
)

backends/arm/operator_support/tosa_supported_operators.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from executorch.backends.arm.tosa_specification import TosaSpecification
1212
from executorch.exir.dialects._ops import ops as exir_ops
1313
from torch.fx.passes.operator_support import OperatorSupportBase
14+
from typing import Type
1415

1516

1617
class SupportedTOSAOperatorCheck:
@@ -31,7 +32,9 @@ def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool
3132

3233

3334
# container for all SupportedTosaOperatorCheck classes
34-
_tosa_spec_dicts: dict[TosaSpecification, dict[str, SupportedTOSAOperatorCheck]] = {
35+
_tosa_spec_dicts: dict[
36+
TosaSpecification, dict[str, Type[SupportedTOSAOperatorCheck]]
37+
] = {
3538
TosaSpecification.create_from_string("TOSA-0.80.0+BI"): {},
3639
TosaSpecification.create_from_string("TOSA-0.80.0+MI"): {},
3740
}

0 commit comments

Comments
 (0)