Skip to content

Commit c22d879

Browse files
author
Martin Lindström
committed
Arm Backend: Remove unused param in aot_arm_compiler.py
Signed-off-by: Martin Lindström <[email protected]> Change-Id: I756f7daf44097dcb9c91a4fdf27cacf979a2e871
1 parent d069d65 commit c22d879

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -671,7 +671,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
671671

672672

673673
def quantize_model(
674-
exported_program, args, model: torch.nn.Module, example_inputs, compile_spec
674+
args, model: torch.nn.Module, example_inputs, compile_spec
675675
):
676676
model_int8 = quantize(
677677
model,
@@ -704,7 +704,7 @@ def to_edge_TOSA_delegate(
704704
model_int8 = None
705705
if args.quantize:
706706
model_int8, exported_program = quantize_model(
707-
exported_program, args, model, example_inputs, compile_spec
707+
args, model, example_inputs, compile_spec
708708
)
709709
model = model_int8
710710

@@ -740,7 +740,7 @@ def to_edge_no_delegate(exported_program, args, model: torch.nn.Module, example_
740740
args.memory_mode,
741741
)
742742
model, exported_program = quantize_model(
743-
exported_program, args, model, example_inputs, compile_spec
743+
args, model, example_inputs, compile_spec
744744
)
745745
model_int8 = model
746746

0 commit comments

Comments
 (0)