Skip to content

Commit 3ec8d25

Browse files
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 changed

    0 file changed

    +0
    -0
    lines changed

    0 commit comments

    Comments
     (0)