Arm backend: Create ethosu directory (pytorch#11849)

Differential Revision: D77349641

Pull Request resolved: pytorch#12338

Author: digantdesai

backends/arm/TARGETS

@@ -1,10 +1,20 @@
 # @noautodeps
 load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+python_library(
+    name = "ethosu_partitioner",
+    srcs = [
+        "ethosu/__init__.py",
+        "ethosu/backend.py",
+        "ethosu/partitioner.py"
+    ],
+    deps = [
+        ":arm_partitioner",
+    ]
+)
 python_library(
     name = "arm_partitioner",
     srcs = [
-        "ethosu_backend.py",
-        "ethosu_partitioner.py",
         "tosa_backend.py",
         "tosa_partitioner.py",
         "vgf_backend.py",

backends/arm/ethosu/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 2025 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+#
+# pyre-unsafe
+
+from .backend import EthosUBackend  # noqa: F401
+from .partitioner import EthosUPartitioner  # noqa: F401
+
+__all__ = [
+    "EthosUBackend",
+    "EthosUPartitioner",
+]

backends/arm/ethosu_backend.py renamed to backends/arm/ethosu/backend.py

@@ -10,7 +10,7 @@
 from executorch.backends.arm.arm_backend import (
     is_ethosu,
 )  # usort: skip
-from executorch.backends.arm.ethosu_backend import EthosUBackend
+from executorch.backends.arm.ethosu import EthosUBackend
 from executorch.backends.arm.tosa_partitioner import TOSAPartitioner
 from executorch.exir.backend.compile_spec_schema import CompileSpec
 from executorch.exir.backend.partitioner import DelegationSpec

backends/arm/ethosu_partitioner.py renamed to backends/arm/ethosu/partitioner.py

@@ -41,7 +41,7 @@ python_library(
     deps = [
         ":common",
         "//executorch/backends/xnnpack/test/tester:tester",
-        "//executorch/backends/arm:arm_partitioner",
+        "//executorch/backends/arm:ethosu_partitioner",
         "//executorch/backends/arm/quantizer:lib",
         "//executorch/backends/arm:tosa_mapping",
         "//executorch/devtools/backend_debug:delegation_info",

backends/arm/test/TARGETS

@@ -39,7 +39,7 @@
     is_tosa,
     is_vgf,
 )
-from executorch.backends.arm.ethosu_partitioner import EthosUPartitioner
+from executorch.backends.arm.ethosu import EthosUPartitioner
 from executorch.backends.arm.quantizer import (
     EthosUQuantizer,
     get_symmetric_quantization_config,

backends/arm/test/tester/arm_tester.py

@@ -23,7 +23,7 @@ The example below demonstrates the lowering processs of a MobileNet V2 model fro
 ```python
 import torch
 from executorch.backends.arm.arm_backend import ArmCompileSpecBuilder
-from executorch.backends.arm.ethosu_partitioner import EthosUPartitioner
+from executorch.backends.arm.ethosu import EthosUPartitioner
 from executorch.backends.arm.quantizer.arm_quantizer import (
     EthosUQuantizer,
     get_symmetric_quantization_config,

docs/source/backends-arm-ethos-u.md

@@ -24,7 +24,7 @@
     is_tosa,
     is_vgf,
 )
-from executorch.backends.arm.ethosu_partitioner import EthosUPartitioner
+from executorch.backends.arm.ethosu import EthosUPartitioner
 from executorch.backends.arm.quantizer import (
     EthosUQuantizer,
     get_symmetric_quantization_config,

examples/arm/aot_arm_compiler.py

@@ -138,7 +138,7 @@
    "outputs": [],
    "source": [
     "import os\n",
-    "from executorch.backends.arm.ethosu_partitioner import EthosUPartitioner\n",
+    "from executorch.backends.arm.ethosu import EthosUPartitioner\n",
     "from executorch.exir import (\n",
     "    EdgeCompileConfig,\n",
     "    ExecutorchBackendConfig,\n",