We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent df4c12e commit 402f421Copy full SHA for 402f421
backends/cadence/aot/compiler.py
@@ -217,6 +217,7 @@ def _lower_ep_to_edge(
217
expo_program: ExportedProgram,
218
dump_graphs: bool = False,
219
constant_methods: Optional[dict[str, object]] = None,
220
+ core_aten_exceptions: Optional[list[torch._ops.OpOverload]] = None,
221
) -> EdgeProgramManager:
222
"""
223
Lower an ExportedProgram to an EdgeProgramManager (in edge IR).
@@ -237,7 +238,8 @@ def _lower_ep_to_edge(
237
238
torch.ops.aten.unfold.default,
239
torch.ops.aten.angle.default,
240
torch.ops.aten.rms_norm.default,
- ],
241
+ ]
242
+ + (core_aten_exceptions or []),
243
),
244
constant_methods=constant_methods,
245
preserve_ops=(torch.ops.aten.rms_norm.default,),
0 commit comments