Skip to content

Commit c755d04

Browse files
authored
Fix calibration data propagation
Differential Revision: D75532269 Pull Request resolved: #11195
1 parent 86ba7e7 commit c755d04

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)