Commit ac67efc
Add preserve_ops to EdgeCompileConfig (#12546)
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/)1 parent f5405a6 commit ac67efc
1 file changed
+4
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
784 | 784 | | |
785 | 785 | | |
786 | 786 | | |
| 787 | + | |
787 | 788 | | |
| 789 | + | |
| 790 | + | |
| 791 | + | |
788 | 792 | | |
789 | 793 | | |
790 | 794 | | |
| |||
0 commit comments