Commit 3ec8d25
committed
Update on "[Executorch][Export/ Recipes] Modify pre_edge_transform_passes, edge_transform_passes definition to take exported program and method name."
While adding QNN recipes, i realized there are gaps in pre edge and edge transform passes in the recipe logic. Where the transform passes need exported program to resolve them dynamically at runtime. Changes made:
- `pre_edge_transform_passes` is now renamed to `aten_transform_passes`
- `aten_transform_passes`: now accepts list of transformation functions which takes in (method_name, ExportedProgram) and gives back ExportedProgram
- `aten_transform_passes: Optional[List[Callable[[str, ExportedProgram], ExportedProgram]]]`
- `edge_transform_passes`: Now this is a list of callables that resolves to a list of passes, callables is needed because some of the passes might need ExportedProgram which is only available during execution and not when recipe is created.
- `edge_transform_passes: None | List[Callable[[str, ExportedProgram], List[PassType]]]`
Differential Revision: [D81730890](https://our.internmc.facebook.com/intern/diff/D81730890/)
[ghstack-poisoned]1 parent b9dd175 commit 3ec8d25
File tree
0 file changed
+0
-0
lines changed0 file changed
+0
-0
lines changed
0 commit comments