Skip to content

Commit 23ad4eb

Browse files
committed
Arm backend: Fix for non quantized TOSA delegation in aot_arm_compiler
Signed-off-by: Zingo Andersen <[email protected]> Change-Id: I1581c1e8d578418105cdfc3b269fbbd84b0ea641
1 parent 13b5605 commit 23ad4eb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -624,6 +624,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
624624

625625

626626
def to_edge_TOSA_delegate(
627+
exported_program,
627628
args,
628629
model: torch.nn.Module,
629630
):
@@ -687,7 +688,7 @@ def to_edge_TOSA_delegate(
687688
# Quantize if required
688689
model_int8 = None
689690
if args.delegate:
690-
model_int8, edge = to_edge_TOSA_delegate(args, model)
691+
model_int8, edge = to_edge_TOSA_delegate(exported_program, args, model)
691692
else:
692693
edge = to_edge_transform_and_lower(
693694
exported_program,

0 commit comments

Comments
 (0)