Create an initital recipe for xnnpack optimization

.ci/scripts/gather_test_models.py

@@ -9,7 +9,8 @@
 import os
 from typing import Any
 
-from examples.models import MODEL_NAME_TO_MODEL, MODEL_NAME_TO_OPTIONS
+from examples.models import MODEL_NAME_TO_MODEL
+from examples.recipes.xnnpack_optimization import MODEL_NAME_TO_OPTIONS
 
 BUILD_TOOLS = [
     "buck2",

examples/backend/targets.bzl

@@ -22,7 +22,7 @@ def define_common_targets():
         ],
         deps = [
             "//executorch/backends/xnnpack/partition:xnnpack_partitioner",
-            "//executorch/examples/models:models",
+            "//executorch/examples/recipes/xnnpack_optimization:models",
             "//executorch/examples/quantization:quant_utils",
             "//executorch/exir:lib",
             "//executorch/exir/backend:backend_api",

examples/backend/xnnpack_examples.py

@@ -16,8 +16,9 @@
 )
 from executorch.exir.backend.backend_api import to_backend
 
-from ..models import MODEL_NAME_TO_MODEL, MODEL_NAME_TO_OPTIONS
+from ..models import MODEL_NAME_TO_MODEL
 from ..quantization.utils import quantize
+from ..recipes.xnnpack_optimization import MODEL_NAME_TO_OPTIONS
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)

examples/models/__init__.py

@@ -4,6 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-from .models import MODEL_NAME_TO_MODEL, MODEL_NAME_TO_OPTIONS
+from .models import MODEL_NAME_TO_MODEL
 
-__all__ = [MODEL_NAME_TO_MODEL, MODEL_NAME_TO_OPTIONS]
+__all__ = [MODEL_NAME_TO_MODEL]

examples/models/models.py

@@ -142,17 +142,3 @@ def gen_resnet50_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
     "resnet18": gen_resnet18_model_and_inputs,
     "resnet50": gen_resnet50_model_and_inputs,
 }
-
-
-@dataclass
-class OptimizationOptions(object):
-    quantization: bool
-    xnnpack_delegation: bool
-
-
-MODEL_NAME_TO_OPTIONS = {
-    "linear": OptimizationOptions(True, True),
-    "add": OptimizationOptions(True, True),
-    "add_mul": OptimizationOptions(True, True),
-    "mv2": OptimizationOptions(True, True),
-}

examples/quantization/example.py

@@ -26,9 +26,8 @@
 )
 
 from ..export.export_example import export_to_pte
-
-from ..models import MODEL_NAME_TO_MODEL, MODEL_NAME_TO_OPTIONS
-
+from ..models import MODEL_NAME_TO_MODEL
+from ..recipes.xnnpack_optimization import MODEL_NAME_TO_OPTIONS
 from .utils import quantize
 
 

examples/recipes/xnnpack_optimization/TARGETS

@@ -0,0 +1,12 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+python_library(
+    name = "models",
+    srcs = [
+        "__init__.py",
+        "models.py",
+    ],
+    deps = [
+        "//executorch/examples/models:models",  # @manual
+    ],
+)

examples/recipes/xnnpack_optimization/__init__.py

@@ -0,0 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from .models import MODEL_NAME_TO_OPTIONS
+
+__all__ = [MODEL_NAME_TO_OPTIONS]

examples/recipes/xnnpack_optimization/models.py

@@ -0,0 +1,21 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from dataclasses import dataclass
+
+
+@dataclass
+class OptimizationOptions(object):
+    quantization: bool
+    xnnpack_delegation: bool
+
+
+MODEL_NAME_TO_OPTIONS = {
+    "linear": OptimizationOptions(True, True),
+    "add": OptimizationOptions(True, True),
+    "add_mul": OptimizationOptions(True, True),
+    "mv2": OptimizationOptions(True, True),
+}