Skip to content

Commit a0c7570

Browse files
mcremon-metafacebook-github-bot
authored andcommitted
Fix calibration data propagation
Summary: `quantize_and_export_to_executorch` is not using the variable. Propagate it properly to `quantize_and_export_to_edge` and then `quantize_pt2` Differential Revision: D75532269
1 parent d8ac866 commit a0c7570

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/cadence/aot/compiler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,7 @@ def quantize_and_export_to_edge(
275275
quantizer: Optional[CadenceQuantizer] = None,
276276
dump_graphs: bool = False,
277277
constant_methods: Optional[dict[str, object]] = None,
278+
calibration_data: Optional[list[tuple[object, ...]]] = None,
278279
) -> EdgeProgramManager:
279280
"""
280281
Trace, quantize and lower a model/inputs pair to edge IR.
@@ -283,6 +284,7 @@ def quantize_and_export_to_edge(
283284
model,
284285
inputs,
285286
quantizer=quantizer,
287+
calibration_data=calibration_data,
286288
dump_graphs=dump_graphs,
287289
)
288290

0 commit comments

Comments
 (0)