You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
1. Add `preserve_ops` to `EdgeCompileConfig`
2. Remove preserved ops from the decomposition table in `to_edge`.
3. Add checks to the verifier ensuring that preserved ops do not have mutations or views.
4. Update 'core_aten_exception_list' to be 'preserved_ops' in `to_edge_transform_and_lower`.
Context/Usage
**core_aten_ops_exception_list**
- Contains operators that are missing a decomposition to core aten.
- Exclude these so that verification can still be run on the rest of the graph.
- Ideally, this list should be empty.
**preserve_ops**
- Contains operators that the user specifically does not want decomposed.
- Must be aten; custom ops are ignored by verifier.
Edge case:
- If an aten operator does not have a decomp, and the user specifically wants it to be preserved, put it in preserve_ops rather than core_aten_ops_exception_list.
Differential Revision: [D78298749](https://our.internmc.facebook.com/intern/diff/D78298749/)
[ghstack-poisoned]
0 commit comments