Skip to content
13 changes: 4 additions & 9 deletions backends/arm/test/TARGETS
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ python_library(
srcs = ["runner_utils.py"],
deps = [
":conftest",
"//executorch/backends/xnnpack/test/tester:tester",
"//executorch/backends/arm:arm_backend",
"//executorch/exir:lib",
"//executorch/exir/backend:compile_spec_schema",
Expand All @@ -31,10 +30,7 @@ python_library(
srcs = ["common.py"],
deps = [
":runner_utils",
"//executorch/backends/xnnpack/test/tester:tester",
"//executorch/backends/arm:arm_backend",
"//executorch/exir:lib",
"//executorch/exir/backend:compile_spec_schema",
"//executorch/backends/arm:tosa_specification",
"fbsource//third-party/pypi/pytest:pytest",
]
)
Expand All @@ -43,12 +39,11 @@ python_library(
name = "arm_tester",
srcs = glob(["tester/*.py"]),
deps = [
":runner_utils",
":common",
"//executorch/backends/arm:tosa_mapping",
"//executorch/backends/arm:tosa_specification",
"//executorch/backends/arm/quantizer:arm_quantizer",
"//executorch/backends/xnnpack/test/tester:tester",
"//executorch/backends/arm:arm_partitioner",
"//executorch/backends/arm/quantizer:arm_quantizer",
"//executorch/backends/arm:tosa_mapping",
"//executorch/devtools/backend_debug:delegation_info",
"fbsource//third-party/pypi/tabulate:tabulate",
]
Expand Down
11 changes: 8 additions & 3 deletions backends/arm/test/targets.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,16 @@ load("@bazel_skylib//lib:paths.bzl", "paths")

def define_arm_tests():
# TODO Add more tests
test_files = native.glob(["passes/test_*.py"])
test_files = []

# Passes
test_files += native.glob(["passes/test_*.py"])
# https://github.com/pytorch/executorch/issues/8606
test_files.remove("passes/test_ioquantization_pass.py")

# Operators
test_files += native.glob(["ops/test_linear.py"])

TESTS = {}

for test_file in test_files:
Expand All @@ -26,8 +31,8 @@ def define_arm_tests():
"//executorch/kernels/quantized:custom_ops_generated_lib",
],
deps = [
":arm_tester",
":conftest",
"//executorch/backends/arm/test:arm_tester",
"//executorch/backends/arm/test:conftest",
"//executorch/exir:lib",
"fbsource//third-party/pypi/pytest:pytest",
"fbsource//third-party/pypi/parameterized:parameterized",
Expand Down
Loading