diff --git a/backends/arm/test/TARGETS b/backends/arm/test/TARGETS index 58239913e29..38a4ec18787 100644 --- a/backends/arm/test/TARGETS +++ b/backends/arm/test/TARGETS @@ -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", @@ -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", ] ) @@ -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", ] diff --git a/backends/arm/test/targets.bzl b/backends/arm/test/targets.bzl index 3335dc958cd..e97b46cb977 100644 --- a/backends/arm/test/targets.bzl +++ b/backends/arm/test/targets.bzl @@ -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: @@ -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",