From 60a20cc195feeb03f0d29eed6494bbff5dd280ba Mon Sep 17 00:00:00 2001 From: Matthias Cremon Date: Wed, 28 May 2025 09:50:42 -0700 Subject: [PATCH] Fix calibration data propagation (#11195) Summary: As titled. Differential Revision: D75532269 --- backends/cadence/aot/compiler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py index 438f07ba15f..b6f213c9d66 100644 --- a/backends/cadence/aot/compiler.py +++ b/backends/cadence/aot/compiler.py @@ -275,6 +275,7 @@ def quantize_and_export_to_edge( quantizer: Optional[CadenceQuantizer] = None, dump_graphs: bool = False, constant_methods: Optional[dict[str, object]] = None, + calibration_data: Optional[list[tuple[object, ...]]] = None, ) -> EdgeProgramManager: """ Trace, quantize and lower a model/inputs pair to edge IR. @@ -283,6 +284,7 @@ def quantize_and_export_to_edge( model, inputs, quantizer=quantizer, + calibration_data=calibration_data, dump_graphs=dump_graphs, )