diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py index cb91c459cfd..560b625e4c0 100644 --- a/backends/cadence/aot/compiler.py +++ b/backends/cadence/aot/compiler.py @@ -278,6 +278,7 @@ def quantize_and_export_to_edge( dump_graphs: bool = False, constant_methods: Optional[dict[str, object]] = None, calibration_data: Optional[list[tuple[object, ...]]] = None, + core_aten_exceptions: Optional[list[torch._ops.OpOverload]] = None, ) -> EdgeProgramManager: """ Trace, quantize and lower a model/inputs pair to edge IR. @@ -294,6 +295,7 @@ def quantize_and_export_to_edge( quantized_model, dump_graphs=dump_graphs, constant_methods=constant_methods, + core_aten_exceptions=core_aten_exceptions, )