Skip to content

Commit 7e869be

Browse files
authored
Call prepare_traced_pt2 instead of prepare_pt2 in get_fake_quant_model
Differential Revision: D88103722 Pull Request resolved: #16033
1 parent c1113d3 commit 7e869be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/cadence/aot/compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ def get_fake_quant_model(
183183
logging.info(program.graph.print_tabular())
184184

185185
# Get prepared graph module
186-
prepared_gm = prepare_pt2(model, inputs, quantizer, dump_graphs=dump_graphs)
186+
prepared_gm = prepare_traced_pt2(program, quantizer, dump_graphs=dump_graphs)
187187

188188
# Calibrate
189189
# If no calibration data is provided, use the inputs

0 commit comments

Comments
 (0)