Update on "[Executorch][Export][3/N] Modularize export recipes"

Addresses (7) in the rfc: #12660

Changes:
1. Add data class called `LoweringRecipe`
2. Modify current xnnpack recipes to use lowering recipes

Fixes: #12933

Differential Revision: [D79120575](https://our.internmc.facebook.com/intern/diff/D79120575/)

[ghstack-poisoned]

Author: abhinaykukkadapu

export/export.py

@@ -196,44 +196,11 @@ def _build_stages(self, stages: List[StageType]) -> Dict[StageType, Stage]:
                     )
                 stage = TorchExportStage(pre_edge_passes)
             elif stage_type == StageType.TO_EDGE_TRANSFORM_AND_LOWER:
-                stage = EdgeTransformAndLowerStage(
-                    partitioners=(
-                        self._lowering_recipe.partitioners
-                        if self._lowering_recipe
-                        else None
-                    ),
-                    transform_passes=(
-                        self._lowering_recipe.edge_transform_passes
-                        if self._lowering_recipe
-                        else None
-                    ),
-                    compile_config=(
-                        self._lowering_recipe.edge_compile_config
-                        if self._lowering_recipe
-                        else None
-                    ),
-                )
+                stage = EdgeTransformAndLowerStage.from_recipe(self._lowering_recipe)
             elif stage_type == StageType.TO_EDGE:
-                stage = ToEdgeStage(
-                    edge_compile_config=(
-                        self._lowering_recipe.edge_compile_config
-                        if self._lowering_recipe
-                        else None
-                    ),
-                )
+                stage = ToEdgeStage.from_recipe(self._lowering_recipe)
             elif stage_type == StageType.TO_BACKEND:
-                stage = ToBackendStage(
-                    partitioners=(
-                        self._lowering_recipe.partitioners
-                        if self._lowering_recipe
-                        else None
-                    ),
-                    transform_passes=(
-                        self._lowering_recipe.edge_transform_passes
-                        if self._lowering_recipe
-                        else None
-                    ),
-                )
+                stage = ToBackendStage.from_recipe(self._lowering_recipe)
             elif stage_type == StageType.TO_EXECUTORCH:
                 stage = ExecutorchStage(self._export_recipe.executorch_backend_config)
             else:

export/stages.py

@@ -14,7 +14,7 @@
 from executorch.exir.backend.backend_api import validation_disabled
 from executorch.exir.program import to_edge, to_edge_transform_and_lower
 from executorch.exir.program._program import _transform
-from executorch.export.recipe import QuantizationRecipe
+from executorch.export.recipe import LoweringRecipe, QuantizationRecipe
 from executorch.export.types import StageType
 from torch import nn
 from torch._export.pass_base import PassType
@@ -168,6 +168,19 @@ def __init__(
         self._transform_passes = transform_passes
         self._compile_config = compile_config
 
+    @classmethod
+    def from_recipe(
+        cls, lowering_recipe: Optional["LoweringRecipe"]
+    ) -> "EdgeTransformAndLowerStage":
+        if lowering_recipe is None:
+            return cls()
+
+        return cls(
+            partitioners=lowering_recipe.partitioners,
+            transform_passes=lowering_recipe.edge_transform_passes,
+            compile_config=lowering_recipe.edge_compile_config,
+        )
+
     @property
     def stage_type(self) -> str:
         return StageType.TO_EDGE_TRANSFORM_AND_LOWER
@@ -369,6 +382,15 @@ def __init__(
         super().__init__()
         self._edge_compile_config = edge_compile_config
 
+    @classmethod
+    def from_recipe(cls, lowering_recipe: Optional["LoweringRecipe"]) -> "ToEdgeStage":
+        if lowering_recipe is None:
+            return cls()
+
+        return cls(
+            edge_compile_config=lowering_recipe.edge_compile_config,
+        )
+
     @property
     def stage_type(self) -> str:
         return StageType.TO_EDGE
@@ -415,6 +437,18 @@ def __init__(
         self._partitioners = partitioners
         self._transform_passes = transform_passes
 
+    @classmethod
+    def from_recipe(
+        cls, lowering_recipe: Optional["LoweringRecipe"]
+    ) -> "ToBackendStage":
+        if lowering_recipe is None:
+            return cls()
+
+        return cls(
+            partitioners=lowering_recipe.partitioners,
+            transform_passes=lowering_recipe.edge_transform_passes,
+        )
+
     @property
     def stage_type(self) -> str:
         return StageType.TO_BACKEND