Skip to content

Commit 402f421

Browse files
authored
Make core aten exception ops modifiable at runtime.
Differential Revision: D75997936 Pull Request resolved: #11497
1 parent df4c12e commit 402f421

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

backends/cadence/aot/compiler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,7 @@ def _lower_ep_to_edge(
217217
expo_program: ExportedProgram,
218218
dump_graphs: bool = False,
219219
constant_methods: Optional[dict[str, object]] = None,
220+
core_aten_exceptions: Optional[list[torch._ops.OpOverload]] = None,
220221
) -> EdgeProgramManager:
221222
"""
222223
Lower an ExportedProgram to an EdgeProgramManager (in edge IR).
@@ -237,7 +238,8 @@ def _lower_ep_to_edge(
237238
torch.ops.aten.unfold.default,
238239
torch.ops.aten.angle.default,
239240
torch.ops.aten.rms_norm.default,
240-
],
241+
]
242+
+ (core_aten_exceptions or []),
241243
),
242244
constant_methods=constant_methods,
243245
preserve_ops=(torch.ops.aten.rms_norm.default,),

0 commit comments

Comments
 (0)